本文从八个方面入手,全方位带你一步步学习 TypeScript 中泛型。
9. 再学泛型 - 图1
在像 C# 和 Java 这样的语言中,可以使用泛型来创建可重用的组件,一个组件可以支持多种类型的数据。这样用户就可以以自己的数据类型来使用组件。
设计泛型的关键目的是在成员之间提供有意义的约束,这些成员可以是:类的实例成员、类的方法、函数参数和函数返回值。

一、泛型是什么?

我们来举个例子,在这个例子中,我们将一步步揭示泛型的作用。首先我们来定义一个通用的 identity 函数,该函数接收一个参数并直接返回它:

  1. function identity (value) {
  2. return value;
  3. }
  4. console.log(identity(1)) // 1

现在,我们将 identity 函数做适当的调整,以支持 TypeScript 的 Number 类型的参数:

  1. function identity (value: Number) : Number {
  2. return value;
  3. }
  4. console.log(identity(1)) // 1

这里 identity 的问题是我们将 Number 类型分配给参数和返回类型,使该函数仅可用于该原始类型。但该函数并不是可扩展或通用的,很明显这并不是我们所希望的。

我们确实可以把 Number 换成 any,我们失去了定义应该返回哪种类型的能力,并且在这个过程中使编译器失去了类型保护的作用
我们的目标是让 identity 函数可以适用于任何特定的类型,为了实现这个目标,我们可以使用泛型来解决这个问题,具体实现方式如下:

  1. function identity <T>(value: T) : T {
  2. return value;
  3. }
  4. console.log(identity<Number>(1)) // 1

对于刚接触 TypeScript 泛型的读者来说,首次看到 语法会感到陌生。
但这没什么可担心的,就像传递参数一样,我们传递了我们想要用于特定函数调用的类型。
9. 再学泛型 - 图2
参考上面的图片,当我们调用 identity(1) ,Number 类型就像参数 1 一样,它将在出现 T 的任何位置填充该类型。
图中 内部的 T 被称为类型变量,它是我们希望传递给 identity 函数的类型占位符,同时它被分配给 value 参数用来代替它的类型:此时 T 充当的是类型,而不是特定的 Number 类型。

其实并不是只能定义一个类型变量,我们可以引入希望定义的任何数量的类型变量。比如我们引入一个新的类型变量 U,用于扩展我们定义的 identity 函数:

  1. function identity <T, U>(value: T, message: U) : T {
  2. console.log(message);
  3. return value;
  4. }
  5. console.log(identity<Number, string>(68, "Semlinker"));

9. 再学泛型 - 图3
除了为类型变量显式设定值之外,一种更常见的做法是使编译器自动选择这些类型,从而使代码更简洁。我们可以完全省略尖括号,比如:

  1. function identity <T, U>(value: T, message: U) : T {
  2. console.log(message);
  3. return value;
  4. }
  5. console.log(identity(68, "Semlinker"));

对于上述代码,编译器足够聪明,能够知道我们的参数类型,并将它们赋值给 T 和 U,而不需要开发人员显式指定它们。
我们来看个例子,感受一下赋值的过程:

  1. function good <T>(value: T): T {
  2. return value
  3. }
  4. // function good <Number>(value: Number): Number { return 42 }
  5. good<Number>(42)
  6. // function good <String>(value: String): String { return 'hello' }
  7. good('hello')
  8. // function good <Number>(value: Number[]): Number[] { return [1, 2, 3] }
  9. good<Number>([1, 2, 3])

如你所见,该函数接收你传递给它的任何类型,使得我们可以为不同类型创建可重用的组件。

现在我们来继续看看之前的 identity 函数:

  1. function identity <T, U>(value: T, message: U) : T {
  2. console.log(message);
  3. return value;
  4. }

如果我们想要返回两种类型的对象该怎么办呢?
针对这个问题,我们有多种方案,其中一种就是使用元组,即为元组设置通用的类型:

  1. function identity <T, U>(value: T, message: U) : [T, U] {
  2. return [value, message];
  3. }
  4. const [val, msg] = identity(123, '456')

虽然使用元组解决了上述的问题,但当输出数量越来越多的时候呢,而且我们还得按元组的顺序解构赋值,有没有其它更好的方案呢?
答案是有的,你可以使用泛型接口。

二、泛型接口

为了解决上面提到的问题,首先让我们创建一个用于的 identity 函数通用 Identities 接口:

  1. interface Identities<V, M> {
  2. value: V,
  3. message: M
  4. }

在上述的 Identities 接口中,我们引入了类型变量 V 和 M,来进一步说明有效的字母都可以用于表示类型变量,之后我们就可以将 Identities 接口作为 identity 函数的返回类型:

  1. function identity <T, U> (value: T, message: U): Identities<T, U> {
  2. console.log(value + ": " + typeof (value));
  3. console.log(message + ": " + typeof (message));
  4. let identities: Identities<T, U> = {
  5. value,
  6. message
  7. };
  8. return identities;
  9. }
  10. console.log(identity(68, "Semlinker"));
  1. 68: number
  2. Semlinker: string
  3. {value: 68, message: "Semlinker"}

泛型除了可以应用在函数和接口之外,它也可以应用在类中,下面我们就来看一下在类中如何使用泛型。

三、泛型类

在类中使用泛型也很简单,我们只需要在类名后面,使用 <T, ...> 的语法定义任意多个类型变量,具体示例如下:

  1. interface GenericInterface<U> {
  2. value: U
  3. getIdentity: () => U
  4. }
  5. class IdentityClass<T> implements GenericInterface<T> {
  6. value: T
  7. constructor(value: T) {
  8. this.value = value
  9. }
  10. getIdentity(): T {
  11. return this.value
  12. }
  13. }
  14. const myNumberClass = new IdentityClass<Number>(68);
  15. console.log(myNumberClass.getIdentity()); // 68
  16. const myStringClass = new IdentityClass<string>("Semlinker!");
  17. console.log(myStringClass.getIdentity()); // Semlinker!

接下来我们以实例化 myNumberClass 为例,来分析一下其调用过程:

  • 在实例化 IdentityClass 对象时,我们传入 Number 类型和构造函数参数值 68;
  • 之后在 IdentityClass 类中,类型变量 T 的值变成 Number 类型;
  • IdentityClass 类实现了 GenericInterface<T>,而此时 T 表示 Number 类型,因此等价于该类实现了 GenericInterface<Number> 接口;
  • 而对于 GenericInterface<U> 接口来说,类型变量 U 也变成了 Number。这里我有意使用不同的变量名,以表明类型值沿链向上传播,且与变量名无关。

泛型类可确保在整个类中一致地使用指定的数据类型。
比如,你可能已经注意到在使用 Typescript 的 React 项目中使用了以下约定:

  1. type Props = {
  2. className?: string
  3. ...
  4. };
  5. type State = {
  6. submitted?: bool
  7. ...
  8. };
  9. class MyComponent extends React.Component<Props, State> {
  10. ...
  11. }

相信看到这里一些读者会有疑问,我们在什么时候需要使用泛型呢?通常在决定是否使用泛型时,我们有以下两个参考标准:

  • 当你的函数、接口或类将处理多种数据类型时
  • 当函数、接口或类在多个地方使用该数据类型时

很有可能你没有办法保证在项目早期就使用泛型的组件,但是随着项目的发展,组件的功能通常会被扩展。这种增加的可扩展性最终很可能会满足上述两个条件,在这种情况下,引入泛型将比复制组件来满足一系列数据类型更干净。
我们将在本文的后面探讨更多满足这两个条件的用例。不过在这样做之前,让我们先介绍一下 Typescript 泛型提供的其他功能。

四、泛型约束

有时我们可能希望限制每个类型变量接受的类型数量,这就是泛型约束的作用。
下面我们来举几个例子,介绍一下如何使用泛型约束。

4.1 确保属性存在

有时候,我们希望类型变量对应的类型上存在某些属性。
这时,除非我们显式地将特定属性定义为类型变量,否则编译器不会知道它们的存在。

一个很好的例子是在处理字符串或数组时,我们会假设 length 属性是可用的。
让我们再次使用 identity 函数并尝试输出参数的长度:

  1. function identity<T>(arg: T): T {
  2. console.log(arg.length); // Error
  3. return arg;
  4. }

在这种情况下,编译器将不会知道 T 确实含有 length 属性,尤其是在可以将任何类型赋给类型变量 T 的情况下。
我们需要做的就是让类型变量 extends 一个含有我们所需属性的接口,比如这样:

  1. interface Length {
  2. length: number;
  3. }
  4. function identity<T extends Length>(arg: T): T {
  5. console.log(arg.length); // 可以获取length属性
  6. return arg;
  7. }

T extends Length 用于告诉编译器,我们支持已经实现 Length 接口的任何类型。
之后,当我们使用不含有 length 属性的对象作为参数调用 identity 函数时,TypeScript 会提示相关的错误信息:

  1. identity(68); // Error
  2. // Argument of type '68' is not assignable to parameter of type 'Length'.(2345)

此外,我们还可以使用 **,** 号来分隔多种约束类型,比如:**<T extends Length, Type2, Type3>**

4.2 检查对象上的键是否存在

泛型约束的另一个常见的使用场景就是检查对象上的键是否存在。
通过 keyof 操作符,我们就可以获取指定类型的所有键,之后我们就可以结合前面介绍的 extends 约束,即限制输入的属性名包含在 keyof 返回的联合类型中。具体的使用方式如下:

  1. function getProperty<T, K extends keyof T>(obj: T, key: K): T[K] {
  2. return obj[key];
  3. }

在以上的 getProperty 函数中,我们通过 K extends keyof T 确保参数 key 一定是对象中含有的键,这样就不会发生运行时错误。
这是一个类型安全的解决方案,与简单调用 let value = obj[key] 不同。

  1. enum Difficulty {
  2. Easy,
  3. Intermediate,
  4. Hard
  5. }
  6. function getProperty<T, K extends keyof T>(obj: T, key: K): T[K] {
  7. return obj[key];
  8. }
  9. let tsInfo = {
  10. name: "Typescript",
  11. supersetOf: "Javascript",
  12. difficulty: Difficulty.Intermediate
  13. }
  14. let difficulty: Difficulty = getProperty(tsInfo, 'difficulty'); // OK
  15. let supersetOf: string = getProperty(tsInfo, 'superset_of'); // Error
  16. // Argument of type '"superset_of"' is not assignable to parameter of type '"difficulty" | "name" | "supersetOf"'.(2345)

很明显通过使用泛型约束,在编译阶段我们就可以提前发现错误,大大提高了程序的健壮性和稳定性。接下来,我们来介绍一下泛型参数默认类型。

五、泛型参数默认类型

在 TypeScript 2.3 以后,我们可以为泛型中的类型参数指定默认类型。
当使用泛型时没有在代码中直接指定类型参数,从实际值参数中也无法推断出类型时,这个默认类型就会起作用。

泛型参数默认类型与普通函数默认值类似,对应的语法很简单,即 <T=Default Type>,对应的使用示例如下:

  1. interface A<T = string> {
  2. name: T;
  3. }
  4. const strA: A = { name: "Semlinker" }; // 所以,这里的 泛型 A 就可以省略泛型参数
  5. const numB: A<number> = { name: 101 };

泛型参数的默认类型遵循以下规则:

  • 有默认类型的类型参数被认为是可选的。
  • 必选的类型参数不能在可选的类型参数后。
  • 如果类型参数有约束,类型参数的默认类型必须满足这个约束。
  • 当指定类型实参时,你只需要指定必选类型参数的类型实参。未指定的类型参数会被解析为它们的默认类型。
  • 如果指定了默认类型,且类型推断无法选择一个候选类型,那么将使用默认类型作为推断结果。
  • 一个被现有类或接口合并的类或者接口的声明可以为现有类型参数引入默认类型。
  • 一个被现有类或接口合并的类或者接口的声明可以引入新的类型参数,只要它指定了默认类型。

六、泛型条件类型

在 TypeScript 2.8 中引入了条件类型,使得我们可以根据某些条件得到不同的类型,这里所说的条件是类型兼容性约束。
尽管以下代码中使用了 extends 关键字,也不一定要强制满足继承关系,而是检查是否满足结构兼容性。
条件类型会以一个条件表达式进行类型关系检测,从而在两种类型中选择其一:

  1. T extends U ? X : Y

以上表达式的意思是:若类型 T 满足 U 的约束时,那么类型是 X,否则为 Y。
在条件类型表达式中,我们通常还会结合 infer 关键字,实现类型抽取:

  1. interface Dictionary<T = any> {
  2. [key: string]: T;
  3. }
  4. type StrDict = Dictionary<string>
  5. type DictMember<T> = T extends Dictionary<infer V> ? V : never
  6. type StrDictMember = DictMember<StrDict> // string

在上面示例中,当类型 T 满足 T extends Dictionary 约束时,我们会使用 infer 关键字声明了一个类型变量 V,并返回该类型,否则返回 never 类型。

在 TypeScript 中,never 类型表示的是那些永不存在的值的类型。 例如, never 类型是那些总是会抛出异常或根本就不会有返回值的函数表达式或箭头函数表达式的返回值类型。 另外,需要注意的是,没有类型是 never 的子类型或可以赋值给 never 类型(除了 never 本身之外)。 即使 any 也不可以赋值给 never。

除了上述的应用外,利用条件类型infer 关键字,我们还可以方便地实现获取 Promise 对象的返回值类型,比如:

  1. async function stringPromise() {
  2. return "Hello, Semlinker!";
  3. }
  4. interface Person {
  5. name: string;
  6. age: number;
  7. }
  8. async function personPromise() {
  9. return { name: "Semlinker", age: 30 } as Person;
  10. }
  11. type PromiseType<T> = (args: any[]) => Promise<T>;
  12. type UnPromisify<T> = T extends PromiseType<infer U> ? U : never;
  13. type extractStringPromise = UnPromisify<typeof stringPromise>; // string
  14. type extractPersonPromise = UnPromisify<typeof personPromise>; // Person

七、泛型工具类型

为了方便开发者 TypeScript 内置了一些常用的工具类型,比如 Partial、Required、Readonly、Record 和 ReturnType 等。

7.1 Partial

Partial<T> 的作用就是将某个类型里的属性全部变为可选项 ?

  1. type Partial<T> = {
  2. [P in keyof T]?: T[P];
  3. };

在以上代码中,首先通过 keyof T 拿到 T 的所有属性名,然后使用 in 进行遍历,将值赋给 P,最后通过 T[P] 取得相应的属性值。中间的 ? 号,用于将所有属性变为可选。

  1. interface Todo {
  2. title: string;
  3. description: string;
  4. }
  5. function updateTodo(todo: Todo, fieldsToUpdate: Partial<Todo>) {
  6. return { ...todo, ...fieldsToUpdate };
  7. }
  8. const todo1 = {
  9. title: "organize desk",
  10. description: "clear clutter"
  11. };
  12. const todo2 = updateTodo(todo1, {
  13. description: "throw out trash"
  14. });

7.2 Record

Record<K extends keyof any, T> 的作用是将 K 中所有的属性的值转化为 T 类型。

  1. type Record<K extends keyof any, T> = {
  2. [P in K]: T;
  3. };
  1. interface PageInfo {
  2. title: string;
  3. }
  4. type Page = "home" | "about" | "contact";
  5. const x: Record<Page, PageInfo> = {
  6. about: { title: '111' },
  7. contact: { title: '222' },
  8. home: { title: "333" }
  9. };

7.3 Pick

Pick<T, K extends keyof T> 的作用是将某个类型中的子属性挑出来,变成包含这个类型部分属性的子类型。

  1. type Pick<T, K extends keyof T> = {
  2. [P in K]: T[P];
  3. };
  1. interface Todo {
  2. title: string;
  3. description: string;
  4. completed: boolean;
  5. }
  6. type TodoPreview = Pick<Todo, "title" | "completed">;
  7. const todo: TodoPreview = {
  8. title: "Clean room",
  9. completed: false
  10. };

7.4 Exclude

Exclude<T, U> 的作用是将 T 类型中属于 U 类型的移除掉。

  1. type Exclude<T, U> = T extends U ? never : T;
  1. type T0 = Exclude<"a" | "b" | "c", "a">; // "b" | "c"
  2. type T1 = Exclude<"a" | "b" | "c", "a" | "b">; // "c"
  3. type T2 = Exclude<string | number | (() => void), Function>; // string | number

7.5 ReturnType

ReturnType<T> 的作用是用于获取函数 T 的返回类型。

  1. type ReturnType<T extends (...args: any) => any>
  2. = T extends (...args: any) => infer R ? R : any;
  1. type T0 = ReturnType<() => string>; // string
  2. type T1 = ReturnType<(s: string) => void>; // void
  3. type T2 = ReturnType<<T>() => T>; // {}
  4. type T3 = ReturnType<<T extends U, U extends number[]>() => T>; // number[]
  5. type T4 = ReturnType<any>; // any
  6. type T5 = ReturnType<never>; // any
  7. type T6 = ReturnType<string>; // Error
  8. type T7 = ReturnType<Function>; // Error

简单介绍了泛型工具类型,最后我们来介绍如何使用泛型来创建对象。

八、使用泛型来创建对象

有时,泛型类可能需要基于传入的泛型 T 来创建其类型相关的对象。比如:

  1. class FirstClass {
  2. id: number | undefined;
  3. }
  4. class SecondClass {
  5. name: string | undefined;
  6. }
  7. class GenericCreator<T> {
  8. create(): T {
  9. return new T(); // 报错:“T”仅表示类型,但在此处却作为值使用。
  10. }
  11. }
  12. const creator1 = new GenericCreator<FirstClass>();
  13. const firstClass: FirstClass = creator1.create();
  14. const creator2 = new GenericCreator<SecondClass>();
  15. const secondClass: SecondClass = creator2.create();

以上代码并不能正常运行,对于以上代码,在 TypeScript v3.9.2 编译器下会报错。
那么如何解决这个问题呢?
根据 TypeScript 文档,为了使 GenericCreator 能够创建 T 类型的对象,我们需要通过其构造函数来引用 T 类型。
对于上述问题,在介绍具体的解决方案前,我们先来介绍一下构造签名。

8.1 构造签名

在 TypeScript 接口中,你可以使用 new 关键字来描述一个构造函数:

  1. interface Point {
  2. new (x: number, y: number): Point;
  3. }

以上接口中的 new (x: number, y: number) 我们称之为构造签名,其语法如下: :::warning ConstructSignature : _new _TypeParametersopt ( ParameterListopt ) TypeAnnotationopt ::: 在上述的构造签名中,TypeParametersopt ParameterListopt TypeAnnotationopt 分别表示:可选的类型参数、可选的参数列表和可选的类型注解。
与该语法相对应的几种常见的使用形式如下:

  1. new C
  2. new C ( ... )
  3. new C < ... > ( ... )

介绍完构造签名,我们再来介绍一个与之相关的概念,即构造函数类型。

8.2 构造函数类型

在 TypeScript 语言规范中这样定义构造函数类型: :::info An object type containing one or more construct signatures is said to be a constructor type. Constructor types may be written using constructor type literals or by including construct signatures in object type literals. ::: 通过规范中的描述信息,我们可以得出以下结论:

  • 包含一个或多个构造签名的对象类型被称为构造函数类型
  • 构造函数类型可以使用构造函数类型字面量包含构造签名的对象类型字面量来编写。

那么什么是构造函数类型字面量呢?
构造函数类型字面量是包含单个构造函数签名的对象类型的简写。具体来说,构造函数类型字面量的形式如下:

  1. new < T1, T2, ... > ( p1, p2, ... ) => R

该形式与以下对象类型字面量是等价的:

  1. { new < T1, T2, ... > ( p1, p2, ... ) : R }

我们来举个实际的例子:

  1. // 构造函数类型字面量
  2. new (x: number, y: number) => Point
  3. // 等价于下面
  4. // 包含构造签名的对象类型字面量
  5. {
  6. new (x: number, y: number): Point;
  7. }

8.3 构造函数类型的应用

在介绍构造函数类型的应用前,我们先来看个例子:

  1. interface Point {
  2. new (x: number, y: number): Point;
  3. x: number;
  4. y: number;
  5. }
  6. // 报错
  7. // 不能将类型“Point2D”分配给类型“Point”。
  8. // 类型“Point2D”提供的内容与签名“new (x: number, y: number): Point”不匹配
  9. class Point2D implements Point {
  10. readonly x: number;
  11. readonly y: number;
  12. constructor(x: number, y: number) {
  13. this.x = x;
  14. this.y = y;
  15. }
  16. }
  17. const point: Point = new Point2D(1, 2);

相信很多刚接触 TypeScript 不久的小伙伴都会遇到上述的问题。
要解决这个问题,我们就需要把对前面定义的 Point 接口进行分离,即把接口的属性构造函数类型进行分离:

  1. interface Point {
  2. x: number;
  3. y: number;
  4. }
  5. interface PointConstructor {
  6. new (x: number, y: number): Point;
  7. }

完成接口拆分之后,除了前面已经定义的 Point2D 类之外,我们又定义了一个 newPoint 工厂函数,该函数用于根据传入的 PointConstructor 类型的构造函数,来创建对应的 Point 对象。

  1. class Point2D implements Point {
  2. readonly x: number;
  3. readonly y: number;
  4. constructor(x: number, y: number) {
  5. this.x = x;
  6. this.y = y;
  7. }
  8. }
  9. function newPoint(
  10. pointConstructor: PointConstructor,
  11. x: number,
  12. y: number
  13. ): Point {
  14. return new pointConstructor(x, y);
  15. }
  16. const point: Point = newPoint(Point2D, 1, 2);

8.4 使用泛型创建对象

了解完构造签名和构造函数类型之后,下面我们来开始解决上面遇到的问题,首先我们需要重构一下 create 方法,具体如下所示:

  1. class GenericCreator<T> {
  2. create<T>(c: { new (): T }): T {
  3. return new c();
  4. }
  5. }

在以上代码中,我们重新定义了 create 成员方法,根据该方法的签名,我们可以知道该方法接收一个参数,其类型是构造函数类型,且该构造函数不包含任何参数,调用该构造函数后,会返回类型 T 的实例。

如果构造函数含有参数的话,比如包含一个 number 类型的参数时,我们可以这样定义 create 方法:

  1. class GenericCreator<T> {
  2. create<T>(c: { new(a: number): T; }, num: number): T {
  3. return new c(num);
  4. }
  5. }

更新完 GenericCreator 泛型类,我们就可以使用下面的方式来创建 FirstClass 和 SecondClass 类的实例:

  1. const creator1 = new GenericCreator<FirstClass>();
  2. const firstClass: FirstClass = creator1.create(FirstClass);
  3. const creator2 = new GenericCreator<SecondClass>();
  4. const secondClass: SecondClass = creator2.create(SecondClass);

完整代码

  1. interface GenericConstructor<T> {
  2. new(): T
  3. }
  4. class FirstClass {
  5. id: number | undefined;
  6. }
  7. class SecondClass {
  8. name: string | undefined;
  9. }
  10. class GenericCreator<T> {
  11. create(constructorInstance: GenericConstructor<T>): T {
  12. return new constructorInstance()
  13. }
  14. }
  15. const creator1 = new GenericCreator<FirstClass>()
  16. const firstClass: FirstClass = creator1.create(FirstClass);
  17. const creator2 = new GenericCreator<SecondClass>()
  18. const secondClass: SecondClass = creator2.create(SecondClass)

参考资料

《一文读懂 TypeScript 泛型及应用》