柯里化是干什么的劲装?首先看看下面這個函數(shù)
let store = (a,b,c) => "這是你的七仔面"
//函數(shù)就好像一個小賣部阻问,一碗七仔面要三張軟妹幣:a,b,c (五個參數(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) //等價于上訴情況
所以這里我們可以看到,函數(shù)柯里化侍筛,是可以用來慢慢湊齊參數(shù)萤皂,延遲函數(shù)的執(zhí)行。(先分期交錢匣椰,后交貨q晌酢)
做個題
現(xiàn)在,我們的目標是實現(xiàn)一個curry 函數(shù)禽笑,達到以下使用效果:
在給到足夠的參數(shù)時入录,執(zhí)行函數(shù)。不夠參數(shù)的時候則返回一個新的curry函數(shù)佳镜。
let curryPlus = curry((a,b,c) => a+b+c) //這里給到一個有三個參數(shù)的函數(shù)
curryPlus(1)(2)(3) //返回 6
curryPlus(1)(2,3) //返回 6
curryPlus(1,2)(3) //返回 6
let x = curryPlus(1)(2) //喂僚稿,怎么才兩個參數(shù)?返回一個curry函數(shù)(已經(jīng)帶了兩個參數(shù))
x(1) //返回 4
x(2) //返回 5
作為一個js很厲害的人蟀伸,這種問題用一行代碼解決不是問題蚀同,問題是我沒那么厲害。啊掏。蠢络。
于是,我們先從土方法說起吧orz
按照curry的用法脖律,原理是返回了一個函數(shù)谢肾,這個函數(shù)的參數(shù)個數(shù)不確定(你可能同時掏出兩張一塊錢),于是我們可以運用不定參數(shù)的寫法:
const curry = (fn) => {
return (...args) =>{ //不定參數(shù)小泉,想給多少給多少
//給錢交貨環(huán)節(jié)
}
}
柯里化函數(shù)需要記住你已經(jīng)給過他的參數(shù)芦疏,如果沒給的話,則默認為一個空數(shù)組:
const curry = (fn微姊,arr=[]) => { //arr數(shù)組用于記錄已有參數(shù)
return (...args) =>{
//給錢交貨環(huán)節(jié)
}
}
接下來每次調(diào)用的時候酸茴,需要檢查參數(shù)是否給夠,如果夠了兢交,則執(zhí)行fn薪捍,沒有的話則返回一個新的curry函數(shù),將現(xiàn)有的參數(shù)塞給他:
const curry = ( fn, arr = []) => {
return (...args) => {
//判斷參數(shù)總數(shù)是否和fn參數(shù)個數(shù)相等
if([...arr, ...args].length === fn.length){
return fn(...arr, ...args) //拓展參數(shù)配喳,調(diào)用fn
}else{
return curry(fn,[...arr, ...args]) //迭代酪穿,傳入現(xiàn)有的所有參數(shù)
}
}
}
到這里,其實我們已經(jīng)實現(xiàn)了curry函數(shù)晴裹。
接下來就是要看看怎么寫才能更簡潔被济。
首先,中間的代碼可以寫成一個立即執(zhí)行函數(shù)涧团,省掉一些...arr, ...args:
const curry = ( fn, arr = []) => {
return (...args) => {
return ( a => { //a是一個數(shù)組
if(a.length === fn.length) {
return fn(...a)
}else{
return curry(fn, a)
}
})([...arr, ...args]) //這里把arr和args攤開成一個數(shù)組賦值給a
}
}
if語句可以縮減成三元表達式只磷,也可以省掉很多字啦:
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和大括號都給省了:
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
當然了预厌,柯里化函數(shù)的主要作用還是延遲執(zhí)行,執(zhí)行的觸發(fā)條件不一定是參數(shù)個數(shù)相等元媚,也可以是其他的條件轧叽,例如參數(shù)個為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) //返回一個函數(shù)
curryPlus(1)(2) //返回一個函數(shù)
//遇到參數(shù)個數(shù)為0的情況才執(zhí)行
curryPlus(1)(2)(4)() //返回7
curryPlus(1,2)(4)() //返回7