浅谈柯里化以及实现原理

265 阅读1分钟

什么是柯里化

在维基百科中,对柯里化的定义是把接受多个参数函数变换成接受一个单一参数(最初函数的第一个参数)的函数,并且返回接受余下的参数而且返回结果的新函数的技术。

柯里化的使用

举一个例子🌰

例如我们需要编写一个求和函数,我们可能会这样写

const getSum = (a, b, c) => {
  return a + b + c;
};

getSum(1,2,3);

我们简单地对这个函数进行柯里化,先利用lodash中的柯里化方法,我们可以这样写

const _ = require("lodash");

const curried = _.curry(getSum);

curried(1)(2)(3);
curried(1, 2)(3);

再举一个例子🌰

我们编写一个匹配函数,查找字符串是否包含数字

// 普通写法
const match = (reg, str) => {
 return str.match(reg);
}

match(/\d+/g, 'nba123')

// 柯里化案例
const curryMath = _.curry(match);
const hasNumber = curryMath(/\d+/g);

hasNumber("nba123")

柯里化是对函数参数的缓存,让函数变得更灵活,粒度更小。

柯里化的实现

我们可以自己实现一下柯里化函数,如下

// 自定义柯里化方法
function myCurried(func) {
  return function carriedFunc(...args) {
    // 判断实参和形参的个数,**func.length** 表示所传入函数的参数个数
    if (args.length < func.length) { 
      return function (...arg1) {
        return carriedFunc(...[...args, ...arg1]);
      };
    }
    return func(...args);
  };
}

// 测试代码
const getSum = (a, b, c) => {
  return a + b + c;
};

const curried = myCurried(getSum);

console.log(curried(1)(2, 3));
console.log(curried(1, 2)(3));
console.log(curried(1, 2, 3));