快速掌握 TypeScript 新語法:Infer Extends

語言: CN / TW / HK

我們知道,TypeScript 支持 infer 來提取類型的一部分,通過模式匹配的方式。

比如元組類型提取最後一個元素的類型:

type Last<Arr extends unknown[]> = 
    Arr extends [...infer rest,infer Ele]
        ? Ele 
        : never;

比如函數提取返回值類型:

type GetReturnType<Func extends Function> = 
    Func extends (...args: any[]) => infer ReturnType 
        ? ReturnType 
        : never;

比如字符串提取一部分,然後替換:

type ReplaceStr<
    Str extends string,
    From extends string,
    To extends string
> = Str extends `${infer Prefix}${From}${infer Suffix}` 
        ? `${Prefix}${To}${Suffix}` : Str;

模式匹配就是通過一個類型匹配一個模式類型,需要提取的部分通過 infer 聲明一個局部變量,這樣就能從局部變量裏拿到提取的類型。

infer 的模式匹配用法還是挺好理解的。

但是 infer 有一個問題,比如這樣:

從 string 數組中提取的元素,默認會推導為 unknown 類型,這就導致了不能直接把它當 string 用:

那怎麼辦呢?

之前的處理方式是這樣的:

加一層判斷,這樣 Last 就推導為 string 類型了。

或者也可以和 string 取交叉類型:

這樣也可以作為 string 來用。

但是我們明明知道這裏就是 string,卻還需要 & string 或者 xxx extends string 來轉換一次,這也太麻煩了。

TS 也知道有這個問題,所以在 4.7 就引入了新語法:infer extends。

現在我們可以這樣寫:

infer 的時候加上 extends 來約束推導的類型,這樣推導出的就不再是 unknown 了,而是約束的類型。

這個語法是 TS 4.7 引入的,在 4.8 又完善了一下。

比如這樣一個類型:

type NumInfer<Str> = 
    Str extends `${infer Num extends number}`
        ? Num
        : never;

在 4.7 的時候推導結果是這樣:

而 4.8 就是這樣了:

也就是説 4.7 的時候推導出的就是 extends 約束的類型,但是 4.8 的時候,如果是基礎類型,會推導出字面量類型。

有了這個語法之後,除了能簡化類型編程的邏輯之外,也能實現一些之前實現不了的功能:

比如提取枚舉的值的類型:

enum Code {
    a = 111,
    b = 222,
    c = "abc"
}

我們都是這樣寫:

但是有的值明明是數字,卻被作為了字符串,所以要再處理一下,轉換成數字類型,這時候就可以用 infer extends 了:

type StrToNum<Str> =
  Str extends `${infer Num extends number}`
    ? Num
    : Str

做完 string 到 number 的轉換,就拿到了我們想要的結果:

這就是 infer extends 的第二個作用。

處理 string 轉 number 之外,也可以轉 boolean、null 等類型:

總結

Typescript 支持 infer 類型,可以通過模式匹配的方式,提取一部分類型返回。

但是 infer 提取出的類型是 unknown,後面用的時候需要類似和 string 取交叉類型,或者 xxx extends string 這樣的方式來轉換成別的類型來用。這樣比較麻煩。

所以 TS 4.7 實現了 infer extends 的語法,可以指定推導出的類型,這樣簡化了類型編程。

而且,infer extends 還可以用來做類型轉換,比如 string 轉 number、轉 boolean 等。

要注意的是,4.7 的時候,推導出的只是 extends 約束的類型,比如 number、boolean,但是 4.8 就能推導出字面量類型了,比如 1、2、true、false 這種。

有了 infer extends,不但能簡化類型編程,還能實現一些之前很難實現的類型轉換。