TypeScript中如何让match函数实现编译时类型穷尽性检查？

可以通过TypeScript的映射类型、字面量联合类型等高级类型特性，实现编译时强制覆盖所有U变体且避免重复case的match函数，以下是具体实现方案：

核心思路 #

1. 用对象字面量替代数组传递cases：对象的键天然对应U的变体，TypeScript可通过映射类型强制键的完整性，且重复键会直接触发编译报错。
2. 定义映射类型ExhaustiveCases：确保cases对象的键完全覆盖U的所有成员，遗漏任何变体都会编译失败。
3. 限制U为可枚举类型：要求U是字符串、数字、符号的字面量联合类型或枚举，这样TypeScript能追踪所有可能的取值。

实现代码 #

// 定义穷尽性检查的映射类型：强制覆盖U的所有变体
type ExhaustiveCases<T, U extends string | number | symbol> = {
  [K in U]: (v: T) => T;
};

function match<T, U extends string | number | symbol>(
  getKey: (v: T) => U,
  cases: ExhaustiveCases<T, U>
): (v: T) => T {
  return (v) => {
    const key = getKey(v);
    // 因cases已覆盖所有U变体，无需判断undefined，直接调用对应处理函数
    return cases[key](v);
  };
}

使用示例 #

场景1：字面量联合类型 #

type Fruit = 'apple' | 'banana' | 'orange';

interface FruitObj {
  type: Fruit;
  weight: number;
}

// 编译通过：覆盖所有Fruit变体
const processFruit = match(
  (obj: FruitObj) => obj.type,
  {
    apple: (obj) => ({ ...obj, weight: obj.weight * 0.9 }),
    banana: (obj) => ({ ...obj, weight: obj.weight * 0.8 }),
    orange: (obj) => ({ ...obj, weight: obj.weight * 0.7 }),
  }
);

// 编译报错：遗漏orange变体
// const processFruitError = match(
//   (obj: FruitObj) => obj.type,
//   {
//     apple: (obj) => ({ ...obj, weight: obj.weight * 0.9 }),
//     banana: (obj) => ({ ...obj, weight: obj.weight * 0.8 }),
//   }
// );

// 编译报错：重复apple键
// const processFruitDuplicate = match(
//   (obj: FruitObj) => obj.type,
//   {
//     apple: (obj) => ({ ...obj, weight: obj.weight * 0.9 }),
//     apple: (obj) => ({ ...obj, weight: obj.weight * 0.8 }),
//     orange: (obj) => ({ ...obj, weight: obj.weight * 0.7 }),
//   }
// );

场景2：枚举类型 #

enum Status {
  Active,
  Inactive,
  Suspended
}

interface User {
  status: Status;
  name: string;
}

const updateUserStatus = match(
  (user: User) => user.status,
  {
    [Status.Active]: (user) => ({ ...user, name: `${user.name} (Active)` }),
    [Status.Inactive]: (user) => ({ ...user, name: `${user.name} (Inactive)` }),
    [Status.Suspended]: (user) => ({ ...user, name: `${user.name} (Suspended)` }),
  }
);

方案优势 #

• 编译时穷尽性检查：遗漏任何U变体都会触发编译错误，彻底避免运行时的"不可达代码"报错。
• 天然避免重复case：对象字面量不允许重复键，重复定义会直接编译报错。
• 类型安全：所有处理函数的参数和返回值类型都被严格约束，避免类型不匹配问题。

内容的提问来源于stack exchange，提问作者jzimmerman