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

場景

最近在設計一些基礎的項目框架設計上的 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()

感嘆

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

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

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

推薦閱讀: