深入了解函数柯里化

curry是一种处理函数的高级技术。它不仅在JavaScript中使用,也在其他语言中使用。

套用是函数的一种转换,将函数从可调用的f(a, b, c)转换为可调用的f(a)(b)(c)。

curry不调用函数。它只是改变了它。

让我们先看一个例子,以便更好地理解我们正在讨论的内容,然后看实际应用程序。

我们将创建一个辅助函数curry(f),它执行对两个参数f的curry。换句话说,对于两个参数f(a, b)的curry(f)将其转换为一个以f(a)(b)的方式运行的函数:

function curry(f) { // curry(f) does the currying transform
  return function(a) {
    return function(b) {
      return f(a, b);
    };
  };
}

// usage
function sum(a, b) {
  return a + b;
}

let curriedSum = curry(sum);

alert( curriedSum(1)(2) ); // 3

如您所见,实现很简单:它只是两个包装器。

curry(func)的结果是一个包装函数(a)。

当像curriedSum(1)那样调用时,参数被保存在词法环境中,并返回一个新的包装器函数(b)。

然后用2作为参数调用这个包装器,并将调用传递给原始的sum。

更高级的套用实现,例如lodash库中,返回一个允许函数被正常或部分调用的包装器:

function sum(a, b) {
  return a + b;
}

let curriedSum = _.curry(sum); // using _.curry from lodash library

alert( curriedSum(1, 2) ); // 3, still callable normally
alert( curriedSum(1)(2) ); // 3, called partially

为了理解这些好处,我们需要一个有价值的现实例子。

例如,我们有日志功能log(date、importance、message)来格式化和输出信息。在实际的项目中,这样的函数有很多有用的特性,比如通过网络发送日志,这里我们只使用alert:

function log(date, importance, message) {
  alert(`[${date.getHours()}:${date.getMinutes()}] [${importance}] ${message}`);
}

对其进行函数柯里化

log = _.curry(log);

日志正常工作后:

log(new Date(), "DEBUG", "some debug"); // log(a, b, c)

也可以使用 柯里化

log(new Date())("DEBUG")("some debug"); // log(a)(b)(c)

现在我们可以很容易地为当前日志创建一个方便的函数:

// logNow will be the partial of log with fixed first argument
let logNow = log(new Date());

// use it
logNow("INFO", "message"); // [HH:mm] INFO message

现在logNow是带有固定第一个参数的日志,换句话说就是“部分应用函数”或简称为“partial”。

我们可以更进一步,为当前调试日志创建一个方便的函数:

let debugNow = logNow("DEBUG");

debugNow("message"); // [HH:mm] DEBUG message

所以:

curry后我们没有丢失任何东西:log仍然可以正常调用。

我们可以很容易地生成部分函数,比如今天的日志。

进阶的柯里化实现

如果您想了解更多细节,这里是我们可以在上面使用的多参数函数的“高级”curry实现。

很短:

function curry(func) {

  return function curried(...args) {
    if (args.length >= func.length) {
      return func.apply(this, args);
    } else {
      return function(...args2) {
        return curried.apply(this, args.concat(args2));
      }
    }
  };

}

案例

function sum(a, b, c) {
  return a + b + c;
}

let curriedSum = curry(sum);

alert( curriedSum(1, 2, 3) ); // 6, still callable normally
alert( curriedSum(1)(2,3) ); // 6, currying of 1st arg
alert( curriedSum(1)(2)(3) ); // 6, full currying

新的curry看起来可能很复杂,但实际上很容易理解。

curry(func)调用的结果是这样的包装器curry:

// func is the function to transform
function curried(...args) {
  if (args.length >= func.length) { // (1)
    return func.apply(this, args);
  } else {
    return function(...args2) { // (2)
      return curried.apply(this, args.concat(args2));
    }
  }
};

当我们运行它时,有两个if执行分支:

如果传入的args count与原始函数的定义(function.length)相同或更多,则只需使用function.apply将调用传递给它。

否则,得到一个部分:我们还没有调用func。相反,将返回另一个包装器,它将重新应用curry,同时提供以前的参数和新的参数。

然后,如果我们再次调用它,我们将得到一个新的部分(如果没有足够的参数),或者最终得到结果。