簡單三行代碼函數實現(xiàn)幾十行Typescript類型推導

更新時間：2023年01月15日 16:41:46 作者：xekin

這篇文章主要為大家介紹了簡單三行代碼函數實現(xiàn)幾十行Typescript類型推導的方案示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪

場景

最近在設計一些基礎的項目框架設計上的 sdk api，比如埋點系統(tǒng)、權限系統(tǒng)之類的，要提供一些便捷的封裝方法給上層使用。于是遇到了這么個場景。

有一個對象常量里，存了一些方法，例如：

    const METHODS = {
        a: () => "a" as const,
        b: () => "b" as const,
        c: () => "c" as const
    }

然后想要封裝這樣一個 hook 例如 useMethod 給上層的 React 上下文使用：

    type MethodKey = keyof typeof METHODS
    function useMethods(keys: MethodKey[]) {
        return keys.map(key => METHODS[key])
    }
    // case
    const [a, b] = useMethods(['a', 'b'])
    // expect to "a"
    a();

一切都簡簡單單，屬于是日常到不能再日常的代碼，可是當我在 IDE 里挪上去一看，這不對勁呀：

我預期這里應該類型直接就是字符串 a 了，怎么會是個聯(lián)合類型？

摸魚吃瓜式排查

我上上下下看了一遍類型推導，發(fā)現(xiàn) keys.map(key => METHODS[key]) 這一句里，key 直接被推導成了 "a" | "b" | "c"。

所以理所當然的結果也是推導成了 "a" | "b" | "c"。

emmm......這還有些麻煩，先單獨寫個類型方法來推導結果試試，遞歸傳入的數組泛型，取出每一次的 key 對應的 method，再組合為數組。

 type MethodValue<K extends MethodKey> = typeof METHODS[K]
 type GetMethodValue<T extends MethodKey[]> = T extends [] 
     ? [] 
     : T extends [infer F extends MethodKey, ...infer Rest extends MethodKey[]]
     ? [MethodValue<F>, ...GetMethodValue<Rest>]
     : never

測試一下：

再將類型回到方法 useMethod 上帶入卻發(fā)現(xiàn)完全不行：

如果強行斷言 map 返回的結果，則直接會被推導為 never 類型

元組大法

其實不難從代碼里看出，之所以無法推導原因有兩點，第一點是在 Typescript 編譯時這個階段，是無法推導這個函數泛型傳參的多種形態(tài)中的 key 是怎樣排序的，其次是在 map 方法中，key 值一直被推導成 "a" | "b" | "c" 導致。

所以如果我用元組作為泛型限定值，倒是可以實現(xiàn)：

type GetMethodValue<T extends (MethodKey | void)[]> = T extends []
     ? []
     : T extends [infer F extends MethodKey, ...infer Rest extends MethodKey[]] 
     ? [MethodValue<F>, ...GetMethodValue<Rest>] 
     : never 
function useMethods<T extends ['a'?, 'b'?, 'c'?]>(keys: T) {
    return keys.filter((key): key is MethodKey => !!key).map((key) => METHODS[key]) as GetMethodValue<T>
}
const [a, b] = useMethods(['a', 'b'])
const valueA = a()

理解到這，我就思考雖然類型不能自動推導出元組的組合排列方式，但是我卻可以寫一個方法來實現(xiàn)推導聯(lián)合類型生成元組。

    type Permutation<T, U = T> = [T] extends [never]
      ? []
      : U extends T
      ? [U, ...Permutation<Exclude<T, U>>]
      : never;
    // expect to ['a', 'b'] | ['b', 'a']
    type value = Permutation<'a' | 'b'>

這是我之前在寫 TypeChallenge 時寫過的方法，這就派上用場了。

直接將 MethodKey 這個聯(lián)合類型解成元組之后限定泛型 T，最后確實也可以成功推導結果。

type Permutation<T, U = T> = [T] extends [never]
  ? []
  : U extends T
  ? [U?, ...Permutation<Exclude<T, U>>]
  : never;
type MethodKey = keyof typeof METHODS
type MethodValue<K extends MethodKey> = typeof METHODS[K]
type GetMethodValue<T extends (MethodKey | void)[]> = T extends []
     ? []
     : T extends [infer F extends MethodKey, ...infer Rest extends MethodKey[]] 
     ? [MethodValue<F>, ...GetMethodValue<Rest>] 
     : never 
const METHODS = {
    a: () => "a" as const,
    b: () => "b" as const,
    c: () => "c" as const
}
function useMethods<T extends Permutation<MethodKey>>(keys: T) {
    return keys.filter((key): key is MethodKey => !!key)
               .map((key) => METHODS[key]) as GetMethodValue<T>
}
const [a, b] = useMethods(['a', 'b'])
const valueA = a()

感嘆

只是一個三行代碼就實現(xiàn)的簡單方法，但要做出準確的結果推導卻需要這么復雜的類型聲明去鋪墊，雖然最后寫出來很爽，但也感嘆作為庫開發(fā)者的一方真是非常不容易，這當中為了類型推導，還增加了冗余的代碼，為了支持元組的可選值，不得不將變量打為可選，從而需要先 filter 后 map 才能保證結果不會出現(xiàn)空值的類型推導。

身為一個前端，在寫 Ts 時時不時就要為幾個簡單結果的推導準確性花上小半天時間，有時候也覺得很不值得，不知道其他語言在類型上是否也有類似的煩惱，也希望 Typescript 團隊能有更好的類型推斷手段演進。

本文最后的解決方案不一定為最佳解決方案，不過作者也在社區(qū)和搜索網站上檢索過答案，最后也沒找到滿意的解答，更多關于Typescript類型推導代碼函數的資料請關注腳本之家其它相關文章！

您可能感興趣的文章: