簡單三行代碼函數實現幾十行Typescript類型推導

Posted on by

場景

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

有一個對象常量裡，存瞭一些方法，例如：

    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 瞭，怎麼會是個聯合類型？

摸魚吃瓜式排查

我上上下下看瞭一遍類型推導，發現 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 上帶入卻發現完全不行：

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

元組大法

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

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

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()

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

    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 這個聯合類型解成元組之後限定泛型 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()

感嘆

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

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

本文最後的解決方案不一定為最佳解決方案，不過作者也在社區和搜索網站上檢索過答案，最後也沒找到滿意的解答，更多關於Typescript類型推導代碼函數的資料請關註WalkonNet其它相關文章！

推薦閱讀：