泛型与约束：提升 TypeScript 数据结构的精度

python落日圆
192 阅读3分钟

在 TypeScript 中，泛型是一种强大的工具，它极大地增强了我们定义的数据结构的复用性和灵活性。通过泛型，我们可以创建出类型安全且高度可配置的代码。然而，泛型所带来的无限可能有时也可能成为问题。本文将探讨如何通过约束来精细化控制泛型的使用，以确保我们的代码既安全又具有表现力。

1. 泛型的力量与问题

泛型允许我们定义函数和数据结构时不限定具体的类型，而是在实际使用时指定。这提供了巨大的灵活性，但随之而来的问题是泛型可能代表的范围太宽。例如，一个简单的泛型数组可以包含任何类型的对象，而这可能并不是我们所期望的。

2. 引入约束：增强泛型的有用性

为了解决这个问题，TypeScript 允许我们对泛型进行约束。通过使用extends关键字，我们可以指定泛型必须继承自某个特定的类型或接口。这样，虽然看似限制了泛型的范围，但实际上是增强了其有用性，因为现在我们可以确保泛型在使用时满足特定的契约。

interface Printable {
  print(): void;
}

function printHousesOrCars<T extends Printable>(arr: T[]): void {
  for (let i = 0; i < arr.length; i++) {
    arr[i].print();
  }
}

class House implements Printable {
  print() {
    console.log("Printing house details");
  }
}

class Car implements Printable {
  print() {
    console.log("Printing car details");
  }
}

printHousesOrCars([new House(), new House()]);
printHousesOrCars([new Car(), new Car()]);

在这个例子中，printHousesOrCars函数接受一个实现了Printable接口的对象数组，并打印它们的详细信息。通过约束泛型T，我们确保了传入的对象数组中的每个元素都实现了print方法。

3. 完全受限的泛型

尽管使用extends可以约束泛型的下限，但这种方法并没有限制泛型的上限。也就是说，我们可以确保泛型至少实现了某些方法或属性，但无法限制它不能拥有其他的方法或属性。为了实现这种上限的约束，我们可以使用 TypeScript 的条件类型和类型别名。

type UpperLimit = {
  name: string;
  sex: boolean;
  age: number;
  print(text: string): void;
};

type LowerLimit = {
  name: string;
};

// 使用条件类型实现上限和下限的约束
type WithinLimits<T> = UpperLimit extends T ? LowerLimit : never;

function test<T extends WithinLimits<T>>(_: T): void {
  console.log("T is: ", _);
}

const _a = { name: "Alice" };
const _b = { name: "Bob", sex: true };
const _c = { name: "Charlie", sex: true, employer: "CompanyX" };

test(_a); // 正确，满足 LowerLimit
test(_b); // 正确，满足 UpperLimit
// test(_c); // 错误，因为 _c 包含了额外的属性 employer

在上述代码中，我们定义了UpperLimit和LowerLimit两个类型，并通过WithinLimits类型别名实现了对泛型T的上限和下限的约束。这样，test函数就可以确保其参数_既满足了LowerLimit的要求，又没有超出UpperLimit的界限。

4. 泛型约束的实际应用

泛型约束在实际开发中非常有用。例如，在定义一个缓存库时，我们可能希望确保所有缓存的键都是字符串类型。通过使用泛型约束，我们可以轻松实现这一点，同时还能确保缓存的值是任何允许的类型。

class MyCache<TKey, TValue> {
  private storage: Map<TKey, TValue> = new Map();

  set(key: TKey, value: TValue): void {
    this.storage.set(key, value);
  }

  get(key: TKey): TValue | undefined {
    return this.storage.get(key);
  }
}

// 约束键的类型为字符串
type StringKeyCache<TValue> = MyCache<string, TValue>;

const stringKeyIntCache = <StringKeyCache<number>>new MyCache();
stringKeyIntCache.set("key1", 100);
console.log(stringKeyIntCache.get("key1")); // 输出 100

5. 结论

泛型是 TypeScript 中一项强大的功能，它提供了代码的复用性和灵活性。然而，泛型的无限可能性有时也需要被适当地约束，以确保代码的安全性和可维护性。通过使用extends关键字和条件类型，我们可以对泛型进行精确的约束，从而在保持灵活性的同时，增加代码的类型安全性。掌握泛型约束的技巧，可以使我们的 TypeScript 代码更加健壮和可靠。

avatar
文章
阅读
粉丝