title: TypeScript infer 关键字
author: 剑大瑞
category: TypeScript
tag:
- typescript
date: 2023-07-01 12:46:31
今天我们介绍一个 TypeScript
非常强大的关键字 infer。
最近在练习 type-challenges。虽然在先前的博文中有介绍过 infer
关键字的使用,但是直到我在 type-challenges
中使用的时候,才体会到这个关键字的强大之处。这个的感觉有多强大,打个比喻:
好比你在做体操比赛的时候,人已经飞到半空了,但是不知道接下来该做什么动作?该用头着地还是用脚着地?这个时候你要是知道 infer
关键字怎么用,那么它会给你答案!就是这么神奇!
接下来说正事。
首先 infer
关键字必须结合 TypeScript
中的条件类型进行操作,然后结合条件分支进行类型推导。
一个简单示例:
type ArrayElement = T extends (infer Element)[] ? Element : never
ArrayElement
类型可以接受一个数组类型,然后获取数组的元素类型。比如:
type NumList = number[]
type StrList = string[]
type NumEle = ArrayElement // number
type StrEle = ArrayElement // string
就像上面这么操作就可以获取到数组的元素类型。
通过上面的示例,让我们梳理下,infer
关键字做了什么?
-
infer
这个单词本意是 推断、推论 的意思。 - 当在条件类型中使用
infer
关键字的时候,显示的声明了一个新的类型变量Element
- 而
ts
结合条件操作将推断出泛型的元素类型赋值给了Element
整个过程就是这么简单。
infer
关键字不同情况下的使用
对字符串进行解构推断。
在 js 中 string
也是可迭代,解构的。利用这一特性,我们可以使用 infer
关键字获取到 string
类型的不同位置的字符串类型。
比如我们需要一个可以获取字符串类型的首个字符的类型工具类型 GetFirstStr
。
type Name = 'Da Rui'
type Result = GetFirstStr // 期望获取 D
期望 Result = 'D'
。则使用 infer 结合条件类型操作可以实现为:
type GetFirstStr = T extends `${infer R}${infer O}` ? R : never
type Result = GetFirstStr // D
利用此特性,我们可以轻松实现一个可以将字符串类型转为首字母大写的工具类型 Capitalize
:
type Darui = Capitalize // 期望得到的结果 Da Rui
我们只需将面的 GetFirstStr
稍作修改就可以,在推断获取到第一个字符后,对字符使用工具类型 Uppercase
即可:
type Capitalize = T extends `${infer F}${infer O}` ? `${Uppercase}${O}` : S;
type Darui = Capitalize // Da Rui
你可以在 playground 中试一试。就是这么巧妙。
类似的 type-challenges:
- TrimLeft:https://github.com/type-challenges/type-challenges/blob/main/questions/00106-medium-trimleft/README.md
- Tirm: https://github.com/type-challenges/type-challenges/blob/main/questions/00108-medium-trim/README.md
- Replace: https://github.com/type-challenges/type-challenges/blob/main/questions/00116-medium-replace/README.md
- StartWith: https://github.com/type-challenges/type-challenges/blob/main/questions/02688-medium-startswith/README.md
- EndWith: https://github.com/type-challenges/type-challenges/blob/main/questions/02693-medium-endswith/README.md
对数组进行推断
在刚开始的时候我们通过 ArrayElement
来获取数组元素类型。
infer
的强大之处在于可以让你获取数组类型任意索引位置的元素类型。
实现一个 FirstEle
获取第一个索引位置的元素类型:
type FirstEle = T extends [infer F, ...unknown[]] ? typeof F : never
type Result = FirstEle // 1
type Result2 = FirstEle // true
同样的实现 LastEle
type LastEle = T extends [...unknown[], infer L] ? typeof L : never
实现一个去头去尾 MiddleEles
type MiddleEles = T extends [infer F, ...infer M, infer L] ? M: never
type Result = MiddleEles // [2, 3, 4]
type Result2 = MiddleEles // [1, 2, 3]
有了三个基础的思路,我们可以再结合递归,玩出更多花样来:
比如 type-challenges
中的 Flatten
类型,可以实现将嵌套数组类型拍平:
type flatten = Flatten // [1, 2, 3, 4, 5]
实现 Flatten
:
type Flatten = S extends [infer X, ...infer Y] ?
X extends unknown[] ?
Flatten : Flatten
: T
类似的 type-challenges:
- Last of Array: https://github.com/type-challenges/type-challenges/blob/main/questions/00015-medium-last/README.md
- Pop: https://github.com/type-challenges/type-challenges/blob/main/questions/00016-medium-pop/README.md
- Shift: https://github.com/type-challenges/type-challenges/blob/main/questions/03062-medium-shift/README.md
- Reverse:https://github.com/type-challenges/type-challenges/blob/main/questions/03192-medium-reverse/README.md
对函数进行类型推断
在官网中有一个工具类型 ReturnType
可以用于获取函数的返回值类型。
type T0 = ReturnType string>; // string
type T1 = ReturnType void>; // void
type T2 = ReturnType() => T>; // unknown
其实它的实现也非常简单:
type ReturnType any> = T extends (...args: any) => infer R ? R : any
另外我们还可以对函数的参数进行类型推断,实现一个 Parameters
获取函数参数类型
type MyParameters any> = T extends (...args: infer P) => any ? P : never
const foo = (arg1: string, arg2: number): void => {}
type FunctionParamsType = MyParameters // [arg1: string, arg2: number]
我们知道通过 ...
操作符,可以将函数参数转为一个数组类型。如果结合我们上面数组部分的操作,又可以玩出很多花样来,比如实现一个对函数参数类型进行扩展的工具类型:
type Fn = (a: number, b: string) => number
type Result = AppendArgument
// expected be (a: number, b: string, x: boolean) => number
实现:
type AppendArgument any, T> = Fn extends (...args: infer P[]) => infer R ? (...args: [...P, T]) => R : never
类似的 type-challenges:
- Flip Arguments:https://github.com/type-challenges/type-challenges/blob/main/questions/03196-medium-flip-arguments/README.md
总结
在我看来,infer
关键字给了用户一个类型操作窗口,这个窗口可以在进行类型操作时,通过 infer
基于条件类型操作,去动态声明一个新的类型变量。而剩下的工作交给了 ts
编译器,有 ts
编译器去推断出类型变量的类型,然后给用户使用。
最后期望有收获的朋友可以去练一练 type-challenges,并给个 star
,这个仓库确实挺好玩。
欢迎关注我的微信公众号:【coder 狂想曲】
- type-challenges: https://github.com/type-challenges/type-challenges
服务器托管,北京服务器托管,服务器租用 http://www.fwqtg.net
在物联网步进电机人体外骨骼中,脑机接口的应用可以帮助实现对其的智能化控制和调节。 通过将脑机接口技术与步进电机人体外骨骼结合,用户可以通过思维控制外骨骼的动作,实现更为自然和直观的控制方式。具体来说,脑机接口可以从用户的大脑信号中解码出用户的意图,并将其转换为…