当前位置 主页 > 网站技术 > 代码类 >

    ES6如何用一句代码实现函数的柯里化

    栏目:代码类 时间:2020-01-18 18:08

    柯里化是干什么的?首先看看下面这个函数

    let store = (a,b,c) => "这是你的七仔面" 
    //函数就好像一个小卖部,一碗七仔面要三张软妹币:a,b,c (五个参数)
    

    那函数柯里化是什么?就是我们买面吃的过程可能是这样:

    let curryStore = curry(store) //刚刚的小卖部被柯里化了
    
    //最佳状况
    curryStore(1,5,1)// 老板,刚好!不用找了,刚好七块钱。老板: "这是你的七仔面" 
    
    //偶尔出现的情况
    let boss = curryStore(5) //老板,这五块钱你先拿着,我找找有没有一块钱。老板:……
    boss = boss(1) //有了老板,这一块钱你先拿着,我找找还有没有一块钱。。老板:……
    boss(1) //哈哈,终于找到了,给!老板:"这是你的七仔面" 
    
    curryStore(5)(1)(1) //等价于上诉情况
    
    

    所以这里我们可以看到,函数柯里化,是可以用来慢慢凑齐参数,延迟函数的执行。(先分期交钱,后交货!)

    做个题

    现在,我们的目标是实现一个curry 函数,达到以下使用效果:
    在给到足够的参数时,执行函数。不够参数的时候则返回一个新的curry函数。

    let curryPlus = curry((a,b,c) => a+b+c) //这里给到一个有三个参数的函数
    
    curryPlus(1)(2)(3) //返回 6
    curryPlus(1)(2,3) //返回 6
    curryPlus(1,2)(3) //返回 6
    
    let x = curryPlus(1)(2) //喂,怎么才两个参数?返回一个curry函数(已经带了两个参数)
    x(1) //返回 4
    x(2) //返回 5
    
    

    作为一个js很厉害的人,这种问题用一行代码解决不是问题,问题是我没那么厉害。。。

    于是,我们先从土方法说起吧orz

    按照curry的用法,原理是返回了一个函数,这个函数的参数个数不确定(你可能同时掏出两张一块钱),于是我们可以运用不定参数的写法:

    const curry = (fn) => {
      return (...args) =>{ //不定参数,想给多少给多少
        //给钱交货环节
      }
    }
    

    柯里化函数需要记住你已经给过他的参数,如果没给的话,则默认为一个空数组:

    const curry = (fn,arr=[]) => { //arr数组用于记录已有参数
      return (...args) =>{  
        //给钱交货环节
      }
    }
    

    接下来每次调用的时候,需要检查参数是否给够,如果够了,则执行fn,没有的话则返回一个新的curry函数,将现有的参数塞给他:

    const curry = ( fn, arr = []) => {
      return (...args) => { 
    
        //判断参数总数是否和fn参数个数相等
        if([...arr, ...args].length === fn.length){
          return fn(...arr, ...args) //拓展参数,调用fn
        }else{
          return curry(fn,[...arr, ...args]) //迭代,传入现有的所有参数
        }
    
      }
    }
    
    

    到这里,其实我们已经实现了curry函数。

    接下来就是要看看怎么写才能更简洁。
    首先,中间的代码可以写成一个立即执行函数,省掉一些...arr, ...args:

    const curry = ( fn, arr = []) => {
      return (...args) => { 
    
        return ( a => {  //a是一个数组
          if(a.length === fn.length) {
            return fn(...a)
          }else{
            return curry(fn, a)
          }
        })([...arr, ...args]) //这里把arr和args摊开成一个数组赋值给a
    
      }
    }
    
    

    if语句可以缩减成三元表达式,也可以省掉很多字啦:

    const curry = ( fn, arr = []) => {
      return (...args) => { 
    
        return ( a => { 
    
          return a.length === fn.length ? fn(...a) : curry(fn, a)
    
        })([...arr, ...args]) 
    
      }
    }