js如何用一句代碼實(shí)現(xiàn)函數(shù)的柯里化(ES6)


柯里化是干什么的?首先看看下面這個(gè)函數(shù)

let store = (a,b,c) => "這是你的七仔面"  
//函數(shù)就好像一個(gè)小賣部,一碗七仔面要三張軟妹幣:a,b,c (五個(gè)參數(shù))

那函數(shù)柯里化是什么?就是我們買面吃的過程可能是這樣:

let curryStore = curry(store)  //剛剛的小賣部被柯里化了

//最佳狀況
curryStore(1,5,1)// 老板,剛好!不用找了,剛好七塊錢。老板: "這是你的七仔面"  

//偶爾出現(xiàn)的情況
let boss = curryStore(5) //老板,這五塊錢你先拿著,我找找有沒有一塊錢。老板:……
boss = boss(1) //有了老板,這一塊錢你先拿著,我找找還有沒有一塊錢。。老板:……
boss(1) //哈哈,終于找到了,給!老板:"這是你的七仔面"  

curryStore(5)(1)(1) //等價(jià)于上訴情況

所以這里我們可以看到,函數(shù)柯里化,是可以用來慢慢湊齊參數(shù),延遲函數(shù)的執(zhí)行。(先分期交錢,后交貨!)

做個(gè)題

現(xiàn)在,我們的目標(biāo)是實(shí)現(xiàn)一個(gè)curry 函數(shù),達(dá)到以下使用效果:
在給到足夠的參數(shù)時(shí),執(zhí)行函數(shù)。不夠參數(shù)的時(shí)候則返回一個(gè)新的curry函數(shù)。

let curryPlus = curry((a,b,c) => a+b+c)  //這里給到一個(gè)有三個(gè)參數(shù)的函數(shù)

curryPlus(1)(2)(3) //返回 6
curryPlus(1)(2,3) //返回 6
curryPlus(1,2)(3) //返回 6

let x = curryPlus(1)(2) //喂,怎么才兩個(gè)參數(shù)?返回一個(gè)curry函數(shù)(已經(jīng)帶了兩個(gè)參數(shù))
x(1) //返回 4
x(2) //返回 5

作為一個(gè)js很厲害的人,這種問題用一行代碼解決不是問題,問題是我沒那么厲害。。。
于是,我們先從土方法說起吧orz

按照curry的用法,原理是返回了一個(gè)函數(shù),這個(gè)函數(shù)的參數(shù)個(gè)數(shù)不確定(你可能同時(shí)掏出兩張一塊錢),于是我們可以運(yùn)用不定參數(shù)的寫法:

const curry = (fn) => {
    return (...args) =>{  //不定參數(shù),想給多少給多少
        //給錢交貨環(huán)節(jié)
    }
}

柯里化函數(shù)需要記住你已經(jīng)給過他的參數(shù),如果沒給的話,則默認(rèn)為一個(gè)空數(shù)組:

const curry = (fn,arr=[]) => {  //arr數(shù)組用于記錄已有參數(shù)
    return (...args) =>{   
        //給錢交貨環(huán)節(jié)
    }
}

接下來每次調(diào)用的時(shí)候,需要檢查參數(shù)是否給夠,如果夠了,則執(zhí)行fn,沒有的話則返回一個(gè)新的curry函數(shù),將現(xiàn)有的參數(shù)塞給他:

const curry = ( fn, arr = []) => {
    return (...args) => { 

        //判斷參數(shù)總數(shù)是否和fn參數(shù)個(gè)數(shù)相等
        if([...arr, ...args].length === fn.length){
            return fn(...arr, ...args) //拓展參數(shù),調(diào)用fn
        }else{
            return curry(fn,[...arr, ...args]) //迭代,傳入現(xiàn)有的所有參數(shù)
        }

    }
}

到這里,其實(shí)我們已經(jīng)實(shí)現(xiàn)了curry函數(shù)。

接下來就是要看看怎么寫才能更簡潔。
首先,中間的代碼可以寫成一個(gè)立即執(zhí)行函數(shù),省掉一些...arr, ...args:

const curry = ( fn, arr = []) => {
    return (...args) => { 

        return ( a => {   //a是一個(gè)數(shù)組
            if(a.length === fn.length) {
                return fn(...a)
            }else{
                return curry(fn, a)
            }
        })([...arr, ...args])  //這里把a(bǔ)rr和args攤開成一個(gè)數(shù)組賦值給a

    }
}

if語句可以縮減成三元表達(dá)式,也可以省掉很多字啦:

const curry = ( fn, arr = []) => {
    return (...args) => { 

        return ( a => { 

            return a.length === fn.length ?  fn(...a) : curry(fn, a)

        })([...arr, ...args])  

    }
}

最后,既然函數(shù)里啥事都沒干,就只是return的話,不妨使用箭頭函數(shù)最省的寫法 input => output, 把return和大括號(hào)都給省了:

const curry = ( fn, arr = []) => {
    return (...args) => { 
        return ( a => a.length === fn.length? fn(...a) : curry(fn, a))([...arr, ...args])   //先折疊一層
    }
}

再折疊:

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

收工了收工了:

const curry = ( fn, arr = []) => (...args) => ( a => a.length === fn.length? fn(...a) : curry(fn, a))([...arr, ...args]) 
//衣服給您疊好了

拿去試試:

const curry = ( fn, arr = []) => (...args) => ( a => a.length === fn.length? fn(...a) : curry(fn, a))([...arr, ...args])
let curryPlus = curry((a,b,c,d)=>a+b+c+d)

curryPlus(1,2,3)(4) //返回10
curryPlus(1,2)(4)(3) //返回10
curryPlus(1,2)(3,4) //返回10

當(dāng)然了,柯里化函數(shù)的主要作用還是延遲執(zhí)行,執(zhí)行的觸發(fā)條件不一定是參數(shù)個(gè)數(shù)相等,也可以是其他的條件,例如參數(shù)個(gè)為0的情況,那么我們需要對上面curry函數(shù)稍微做修改:

const curry = ( fn, arr = []) => (...args) => ( (a,b) => b.length === 0? fn(...a) : curry(fn, a))([...arr, ...args],[...args])
let curryPlus = curry((...x)=>x.reduce((a,b)=>a+b))

curryPlus(1) //返回一個(gè)函數(shù)
curryPlus(1)(2) //返回一個(gè)函數(shù)

//遇到參數(shù)個(gè)數(shù)為0的情況才執(zhí)行
curryPlus(1)(2)(4)() //返回7
curryPlus(1,2)(4)() //返回7
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

推薦閱讀更多精彩內(nèi)容