JavaScript组合函数

// 普通情况下：
function double (num) {
    return num * 2
}
function square (num) {
    return num * num
}
var count = 10
var result = square(double(count))
console.log(result)
// 组合函数情况下：
function composeFn (d, s) {
    return function (count) {
        return s(d(count))
    }
}
var newFn = composeFn(double, square)
var result2 = newFn(10)
console.log(result2) 

function myCompose (...fns) {
    // 1.首先看一下传入函数的个数，并判断传入的全部参数的类型都是函数类型，不是的话就抛出异常
    var length = fns.length
    for (var i = 0; i < length; i++) {
        if (typeof fns[i] !== 'function') {
            throw new TypeError('Expected arguments are functions')
        }
    } // 2.返回一个compose函数
    return function compose (...args) {
        // 定义一个空数组result，判断参数的数量，如果为空则置为undefined
        var result = args.length > 1 ? [] : undefined
        // 如果args参数的数量大于1执行if中的语句
        if (args.length > 1) {
            for (var i = 0; i < args.length; i++) {
                var _index = 0
                var r = fns[_index].call(this, args[i])
                while (++_index < length) {
                    r = fns[_index].call(this, r)
                    result.push(r)
                }
            }
        } else {
            var index = 0 // 此时的result为第一个函数执行的结果
            result = length ? fns[index].apply(this, args) : args
            // 使用while将第一个函数执行的结果传递给第二个函数，获取到结果，以此传递下去，最终返回result
            while (++index < length) {
                result = fns[index].call(this, result)
            }
        }
        return result
    }
}
function double (num) {
    return num * 2
}
function square (num) {
    return num * num
}
// 传入两个函数参数
var newFn = myCompose(double, square)
// 传入一个参数调用newFn函数
var result1 = newFn(10) 
// 传入多个参数调用newFn函数
var result2 = newFn(10, 20, 30) 
console.log(result1)// 400
console.log(result2)// [ 400, 1600, 3600 ]