携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第31天，点击查看活动详情

前言

mergeData方法主要是将函数元数据合并到数据data中。

参数如下：

• 参数1：目标元数据。
• 参数2：源元数据。

mergeData实现上借助内部方法composeArgs、composeArgsRight和replaceHolders。下面先了解各个内部方法的封装实现。

composeArgs

composeArgs方法可以创建一个数组，该数组是部分应用的参数的组合。

参数说明：

• 参数1:数组类型，表示提供的参数。
• 参数2:数组类型，表示将参数部分化，以在提供的参数之前添加。
• 参数3:数组类型，表示占位符索引。
• 参数4:布尔类型，表示是否指定curried函数的组成。

composeArgs方法方法实现上通过while循环判断给数组项赋值。

源码如下：

var nativeMax = Math.max;

function composeArgs(args, partials, holders, isCurried) {
  var argsIndex = -1,
      argsLength = args.length,
      holdersLength = holders.length,
      leftIndex = -1,
      leftLength = partials.length,
      rangeLength = nativeMax(argsLength - holdersLength, 0),
      result = Array(leftLength + rangeLength),
      isUncurried = !isCurried;

  while (++leftIndex < leftLength) {
    result[leftIndex] = partials[leftIndex];
  }
  while (++argsIndex < holdersLength) {
    if (isUncurried || argsIndex < argsLength) {
      result[holders[argsIndex]] = args[argsIndex];
    }
  }
  while (rangeLength--) {
    result[leftIndex++] = args[argsIndex++];
  }
  return result;
}

composeArgsRight

composeArgsRight方法和composeArgs方法类似，只是参数组合是为内部的partialRight方法量身定制的。

源码如下：

var nativeMax = Math.max;

function composeArgsRight(args, partials, holders, isCurried) {
  var argsIndex = -1,
      argsLength = args.length,
      holdersIndex = -1,
      holdersLength = holders.length,
      rightIndex = -1,
      rightLength = partials.length,
      rangeLength = nativeMax(argsLength - holdersLength, 0),
      result = Array(rangeLength + rightLength),
      isUncurried = !isCurried;

  while (++argsIndex < rangeLength) {
    result[argsIndex] = args[argsIndex];
  }
  var offset = argsIndex;
  while (++rightIndex < rightLength) {
    result[offset + rightIndex] = partials[rightIndex];
  }
  while (++holdersIndex < holdersLength) {
    if (isUncurried || argsIndex < argsLength) {
      result[offset + holders[holdersIndex]] = args[argsIndex++];
    }
  }
  return result;
}

replaceHolders

replaceHolders方法表示用内部占位符替换“数组”中的某一项，并返回其索引的数组。

使用如下：

var arr =[1,2,3,4,5,6]

replaceHolders(arr,'a')
// arr => [1, 2, 3, 4, 5, 6]

replaceHolders(arr)
// arr => [1, 2, 3, 4, 5, 6]

replaceHolders(arr,5)
// arr => [1, 2, 3, 4, '__lodash_placeholder__', 6]

源码如下：

var PLACEHOLDER = '__lodash_placeholder__';

function replaceHolders(array, placeholder) {
  var index = -1,
      length = array.length,
      resIndex = 0,
      result = [];

  while (++index < length) {
    var value = array[index];
    if (value === placeholder || value === PLACEHOLDER) {
      array[index] = PLACEHOLDER;
      result[resIndex++] = index;
    }
  }
  return result;
}

源码实现

import composeArgs from './_composeArgs.js';
import composeArgsRight from './_composeArgsRight.js';
import replaceHolders from './_replaceHolders.js';

var PLACEHOLDER = '__lodash_placeholder__';

var WRAP_BIND_FLAG = 1,
    WRAP_BIND_KEY_FLAG = 2,
    WRAP_CURRY_BOUND_FLAG = 4,
    WRAP_CURRY_FLAG = 8,
    WRAP_ARY_FLAG = 128,
    WRAP_REARG_FLAG = 256;

var nativeMin = Math.min;

function mergeData(data, source) {
  var bitmask = data[1],
      srcBitmask = source[1],
      newBitmask = bitmask | srcBitmask,
      isCommon = newBitmask < (WRAP_BIND_FLAG | WRAP_BIND_KEY_FLAG | WRAP_ARY_FLAG);

  var isCombo =
    ((srcBitmask == WRAP_ARY_FLAG) && (bitmask == WRAP_CURRY_FLAG)) ||
    ((srcBitmask == WRAP_ARY_FLAG) && (bitmask == WRAP_REARG_FLAG) && (data[7].length <= source[8])) ||
    ((srcBitmask == (WRAP_ARY_FLAG | WRAP_REARG_FLAG)) && (source[7].length <= source[8]) && (bitmask == WRAP_CURRY_FLAG));

  // 如果无法合并元数据，直接返回data。
  if (!(isCommon || isCombo)) {
    return data;
  }
  
  // 如果可用，使用源“thisArg”。
  if (srcBitmask & WRAP_BIND_FLAG) {
    data[2] = source[2];
   
   // curry绑定函数时设置。
    newBitmask |= bitmask & WRAP_BIND_FLAG ? 0 : WRAP_CURRY_BOUND_FLAG;
  }
  
  // 组成部分参数。
  var value = source[3];
  if (value) {
    var partials = data[3];
    data[3] = partials ? composeArgs(partials, value, source[4]) : value;
    data[4] = partials ? replaceHolders(data[3], PLACEHOLDER) : source[4];
  }
  // 组成部分正确的参数。
  value = source[5];
  if (value) {
    partials = data[5];
    data[5] = partials ? composeArgsRight(partials, value, source[6]) : value;
    data[6] = partials ? replaceHolders(data[5], PLACEHOLDER) : source[6];
  }
  
  // 如果可用，使用源“argPos”。
  value = source[7];
  if (value) {
    data[7] = value;
  }
  
  // 如果源'ary'较小，则使用‘ary’。
  if (srcBitmask & WRAP_ARY_FLAG) {
    data[8] = data[8] == null ? source[8] : nativeMin(data[8], source[8]);
  }
  
  // 如果未提供源“arity”，则使用源“ariity”。
  if (data[9] == null) {
    data[9] = source[9];
  }
  
  // 使用源“func”和合并位掩码。
  data[0] = source[0];
  data[1] = newBitmask;

  return data;
}

小结

mergeData方法实现上主要通过composeArgs、composeArgsRight、replaceHolders方法，通过判断进行相应的处理，最终返回合并到的数据data。