JS手动实现柯里化(curry)函数

实现效果

const curry_fn = curry(fn);
fn(1, 2, 3) == curry_fn(1)(2)(3);

实现思路

1. 通过闭包的方式储存传入参数
2. 通过函数的length属性获得参数个数
3. 当参数个数不够时直接返回方法
4. 存储的参数个数等于原函数参数个数时执行原函数

• 如果使用ES6参数默认值，length将不等于实际参数个数
• 参数由arguments获取，ES6直接使用rest参数实现

源码实现

function curry(fn) {
    var length = fn.length; //获取原函数的参数个数
    var args = []; // args存储传入参数
    return function curryFn() {
        // 将arguments转换成数组
        var curryArgs = Array.prototype.slice.call(arguments); 
        args = args.concat(curryArgs);
        if (args.length > length) {
            throw new Error('arguments length error')
        }
        // 存储的参数个数等于原函数参数个数时执行原函数
        if (args.length === length) {
            return fn.apply(null, args);
        }
        // 否则继续返回函数
        return curryFn;
    };
}

ES6版

function curry(fn) {
    let length = fn.length;
    let args = [];
    return function curryFn(...curryArgs) {
        args = args.concat(curryArgs);
        if (args.length > length) {
            throw new Error('arguments length error')
        }
        if (args.length === length) {
            return fn(...args);
        }
        return curryFn;
    }
}

无副作用改进版

经评论区微言提醒，做出如下修改

function curry(fn) {
    const length = fn.length;
    const curryFn = (args) => (arg) => {
        const curryArgs = args.concat(arg);
        if (curryArgs.length === length) {
            return fn(...curryArgs);
        }
        return curryFn(curryArgs);
    }
    return curryFn([]);
}

更多文章 yjy5264.github.io

---