【JavaScript】面试手撕柯里化函数

引入

上周我一个学弟，去字节面试实习生。面试官问他有没有用过柯里化，他摇了摇头。

有一说一，确实柯里化在现实中的项目还是用的比较少的。🐶

面试官继续问他是否听说过柯里化，我的老实学弟摇了摇头。半小时面完后，我问学弟面的如何？学弟摇了摇头说，学长，你听说过柯里化吗？

柯里化

定义

首先，我们参照维基百科给出的定义：

在计算机科学中，柯里化 （英语：Currying），又译为卡瑞化 或加里化，是把接受多个参数的函数变换成接受一个单一参数（最初函数的第一个参数）的函数，并且返回接受余下的参数而且返回结果的新函数的技术

举个🌰

我们正常情况下写sum函数是如下的方式

js 复制代码

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

// 调用
sum(1,2)

但是柯里化后是这样实现的

js 复制代码

const sum = function(a) {
  return function(b) {
    return a + b;
  }
}
// 调用
sum(1)(2)

实现

那如果有N个参数呢，以及我要定一个函数能将所有的函数变成柯里化呢？而非仅仅是上面的sum函数

为了解决N个参数的问题，所以我们想到了递归函数。

为了解决将所有的函数变成柯里化，所以我们的入参要加上fn

以下是代码，借鉴了木易杨大佬的代码

js 复制代码

function currying(fn, length) {
    // 第一次调用获取函数 fn 参数的长度，后续调用获取 fn 剩余参数的长度
    length = length || fn.length; 	
    // currying 包裹之后返回一个新函数，接收参数为 ...args
    return function (...args) {			
      // 新函数接收的参数长度是否大于等于 fn 剩余参数需要接收的长度
      return args.length >= length	
          ? fn.apply(this, args) // 满足要求，执行 fn 函数，传入新函数的参数
        : currying(fn.bind(this, ...args), length - args.length) 
      // 不满足要求，递归 currying 函数，新的 fn 为 bind 返回的新函数
      //（bind 绑定了 ...args 参数，未执行），新的 length 为 fn 剩余参数的长度
    }
  }
  
  const sum = function(a,b,c){
    return a + b + c;
  }

const currySum = currying(sum);

console.log(currySum(1,2,3));
console.log(currySum(1)(2)(3));
console.log(currySum(1,2)(3));
// 输出都是 6

快速使用

lodash中的已经带有了curry函数可以使用，使用方式如下：

js 复制代码

const sum = function(a, b, c) {
  return a + b + c;
};
 
var currySum = _.curry(abc);
 
console.log(currySum(1,2,3));
console.log(currySum(1)(2)(3));
console.log(currySum(1,2)(3));
// 输出都是 6

柯里化的作用

感觉柯里化将问题还复杂化了，那柯里化有什么作用呢？

提高自由度

我们可以考虑一个情况，如果对于一个函数，它有3个参数,它的作用是将3个参数相乘返回。如果我突然想乘用2个数的结果呢？是不是使用柯里化就能拿到中间的结果了。

再举个🌰，这个例子借鉴于云中桥大佬

我们工作中会遇到各种需要通过正则检验的需求，比如校验电话号码、校验邮箱、校验身份证号、校验密码等，这时我们会封装一个通用函数checkByRegExp ,接收两个参数，校验的正则对象和待校验的字符串。

js 复制代码

function checkByRegExp(regExp,string) {
    return regExp.test(string);  
}

checkByRegExp(/^1\d{10}$/, '123456789'); // 校验电话号码
checkByRegExp(/^1\d{10}$/, '1234567810'); // 校验电话号码
checkByRegExp(/^(\w)+(.\w+)*@(\w)+((.\w+)+)$/, 'test@163.com'); // 校验邮箱
checkByRegExp(/^(\w)+(.\w+)*@(\w)+((.\w+)+)$/, 'test@qq.com'); // 校验邮箱

然而当要校验的值非常多时，我们前面的正则表达式要一直写着，这会导致代码不美观，我们可以借助柯里化对checkByRegExp进行封装，以简化代码书写，提高代码可读性。

js 复制代码

//进行柯里化
let _check = curry(checkByRegExp);
//生成工具函数，验证电话号码
let checkCellPhone = _check(/^1\d{10}$/);
//生成工具函数，验证邮箱
let checkEmail = _check(/^(\w)+(.\w+)*@(\w)+((.\w+)+)$/);

checkCellPhone('123456789'); // 校验电话号码
checkEmail('test@163.com'); // 校验电话号码

bind函数

其实bind函数也算是提高自由度的例子，不过由于它是官网的函数，所以单独讲下。

这块借鉴于木易杨大佬的文章，在此感谢🙏。

我们先看一下bind的例子

js 复制代码

const obj = {
  name: "小明",
};

const fn = function () {
  console.log(this.name);
}.bind(obj);

这里 bind 用来改变函数执行时候的上下文，但是函数本身并不执行，所以本质上是延迟计算,提高函数的自由度。

我们可以查看一下bind的简单实现，返回了函数，本身也算是一种柯里化吧。

js 复制代码

// 简化实现，完整版实现中的第 2 步
Function.prototype.bind = function (context) {
    var self = this;
    // 第 1 个参数是指定的 this，截取保存第 1 个之后的参数
		// arr.slice(begin); 即 [begin, end]
    var args = Array.prototype.slice.call(arguments, 1); 

    return function () {
        // 此时的 arguments 是指 bind 返回的函数调用时接收的参数
        // 即 return function 的参数，和上面那个不同
      	// 类数组转成数组
        var bindArgs = Array.prototype.slice.call(arguments);
      	// 执行函数
        return self.apply( context, args.concat(bindArgs) );
    }
}

参考资料

「前端进阶」彻底弄懂函数柯里化

深入高阶函数应用之柯里化