美文网首页程序员
lodash-array-difference函数实现

lodash-array-difference函数实现

作者: 心尘_ee7a | 来源:发表于2018-03-18 20:20 被阅读0次

    前言

    今天讲的是difference()differenceBy()differenceWith()函数。

    difference()

    创建一个新数组,这个数组中的值,为第一个数组(array)排除了给定数组(values)中的值。

    我的实现是

    function difference(array, values){
        if(!array || !Array.isArray(array) || array.length<=0){
            return [];
        }
        var except=Array.from(arguments).slice(1);
        if(!except || except.length<=0){
            return array.concat();
        }
       return array.filter(function(value){
            return except.indexOf(value) < 0;
        });
    }
    

    我只是调用了es6语法中Arrayfilter()indexOf(),实现起来比较简单

    lodash的实现是:

    var difference = baseRest(function(array, values) {
      return isArrayLikeObject(array)
        ? baseDifference(array, baseFlatten(values, 1, isArrayLikeObject, true))
        : [];
    });
    

    lodash写的更加的严密。

    一开始会调用baseRest()方法,这个函数主要是对剩余参数的处理,它不用验证参数或者强制输入参数。我的实现是用arguments,而lodash相对复杂,但更通用吧。主要逻辑都在baseDifference()方法里。baseDifference()后面再说。

    不过,我的实现可以做到difference([3, 2, 1],1,2,8)的调用,而lodash貌似还不行,这我还是挺高兴的。

    differenceBy()

    这个方法类似difference() ,额外的接受一个iteratee(迭代器)参数,首先使用迭代器分别迭代arrayvalues中的每个元素,返回的值作为比较值。 结果值是从第一数组中选择。

    我的实现是

    
    function differenceBy(array) {
        if (!array || !Array.isArray(array) || array.length <= 0) {
            return [];
        }
        var except = arguments[1];
        if (!except  || !Array.isArray(except) || except.length <= 0) {
            return array.concat();
        }
        var func = arguments[2];
        if (!func) {
            func = x => x;
        } else if (typeof func === 'string') {
            var key = func;
            func = data => data[key];
        }
        var exc = except.map(function (value) {
            return func(value);
        });
        return array.filter(function (value) {
            return exc.indexOf(func(value)) < 0;
        });
    
    }
    

    类似于与different()实现基本一致,仅仅加了一个函数的转换而已。由于函数的限制,不能再函数后边直接填写参数了,只能包装成数组传递。

    lodash的实现是:

    var differenceBy = baseRest(function(array, values) {
      var iteratee = last(values);
      if (isArrayLikeObject(iteratee)) {
        iteratee = undefined;
      }
      return isArrayLikeObject(array)
        ? baseDifference(array, baseFlatten(values, 1, isArrayLikeObject, true), baseIteratee(iteratee, 2))
        : [];
    });
    

    difference()基本一致。不同的是diffence()默认是调用_.identity()迭代器,baseDifference()最后是传入了baseIteratee(iteratee, 2)迭代器。

    differenceWith()

    这个方法类似difference() ,它额外接受一个 comparator(比较器),它调用比较arrayvalues中的元素。 结果值是从第一数组中选择。comparator 调用参数有两个:(arrVal, othVal)

    我的实现是:

    function differenceWith(array){
        if (!array || !Array.isArray(array) || array.length <= 0) {
            return [];
        }
        var except = arguments[1];
        if (!except  || !Array.isArray(except) || except.length <= 0) {
            return array.concat();
        }
        var comparator = arguments[2];
        if (!comparator || typeof comparator !== 'function') {
            comparator = (arrVal, othVal)=>arrVal===othVal;
        }
        return array.filter(function (value) {
            var flag=true;
            except.forEach(element => {
                if(comparator(value,element)){
                    flag=false;
                    return;
                }
            });
            return flag;
        });
    }
    

    结构也差不多吧,最后filter回调函数里的是遍历要except,当前的valueexcept里的每一个元素做比较,comparator()函数返回true代表相等,false代表不等。如果找到一个相等的,filter就返回false,遍历结束了还没有找到,说明就没有相等的,返回true

    lodash的实现

    var differenceWith = baseRest(function(array, values) {
      var comparator = last(values);
      if (isArrayLikeObject(comparator)) {
        comparator = undefined;
      }
      return isArrayLikeObject(array)
        ? baseDifference(array, baseFlatten(values, 1, isArrayLikeObject, true), undefined, comparator)
        : [];
    });
    
    

    可以看出,和其他两个也都差不多。baseDifference()最后多了一个比较器。我们可以看下具体的实现:

    baseDifference()

    function baseDifference(array, values, iteratee, comparator) {
      var index = -1,
          includes = arrayIncludes,
          isCommon = true,
          length = array.length,
          result = [],
          valuesLength = values.length;
    
      if (!length) {
        return result;
      }
      if (iteratee) {
        values = arrayMap(values, baseUnary(iteratee));
      }
      if (comparator) {
        includes = arrayIncludesWith;
        isCommon = false;
      }
      else if (values.length >= LARGE_ARRAY_SIZE) {
        includes = cacheHas;
        isCommon = false;
        values = new SetCache(values);
      }
      outer:
      while (++index < length) {
        var value = array[index],
            computed = iteratee == null ? value : iteratee(value);
    
        value = (comparator || value !== 0) ? value : 0;
        if (isCommon && computed === computed) {
          var valuesIndex = valuesLength;
          while (valuesIndex--) {
            if (values[valuesIndex] === computed) {
              continue outer;
            }
          }
          result.push(value);
        }
        else if (!includes(values, computed, comparator)) {
          result.push(value);
        }
      }
      return result;
    }
    

    lodash通过这个函数,把三个功能封装到一块了, lodash没有使用原生array的实例方法,而是使用两个while循环的嵌套,实现difference的功能。

    相关文章

      网友评论

        本文标题:lodash-array-difference函数实现

        本文链接:https://www.haomeiwen.com/subject/lqzhqftx.html