使用Omit时TypeScript类型收窄失效的问题及解决方法

问题复现 #

示例代码：

type A = {
  type: 'A';
  foo: string;
  baz: number;
}

type B = {
  type: 'B';
  bar: string;
  baz: number;
}

type Unioned = A | B

直接使用联合类型时，类型收窄正常工作：

function handle(x: Unioned) {
  if (x.type === 'A') {
    console.log(x.foo) // 无报错，x被推断为A类型
  }
}

但使用Omit<Unioned, 'baz'>后，类型收窄失效并报错：

function handle(x: Omit<Unioned, 'baz'>) {
  if (x.type === 'A') {
    console.log(x.foo) // 报错：Property 'foo' does not exist on type 'Omit<Unioned, "baz">'
  }
}

原因分析 #

核心问题出在TypeScript原生Omit工具类型的实现逻辑，以及keyof对联合类型的处理规则：

1. 原生Omit<T, K>的定义是Pick<T, Exclude<keyof T, K>>，依赖keyof获取类型T的所有键。
2. 当T是联合类型时，keyof T返回的是所有联合成员共有的键的交集。对于Unioned = A | B，keyof Unioned的结果是'type' | 'baz'（这两个键是A和B都具备的）。
3. 因此Exclude<keyof Unioned, 'baz'>的结果是'type'，Omit<Unioned, 'baz'>最终等价于Pick<Unioned, 'type'>，也就是{ type: 'A' } | { type: 'B' }——这个类型里根本没有foo或bar属性，自然无法通过类型收窄访问。

我们期望的是对联合类型的每个成员单独应用Omit，得到Omit<A, 'baz'> | Omit<B, 'baz'>，但原生Omit不支持这种分布式处理。

解决办法 #

实现一个支持联合类型的分布式Omit工具类型，利用TypeScript条件类型的分布式特性（当泛型参数是联合类型时，会自动拆分每个成员单独处理，再合并结果）：

// 分布式Omit：对联合类型的每个成员单独应用Omit
type DistributiveOmit<T, K extends PropertyKey> = T extends unknown ? Omit<T, K> : never;

使用这个类型代替原生Omit，类型收窄就能正常工作：

function handle(x: DistributiveOmit<Unioned, 'baz'>) {
  if (x.type === 'A') {
    console.log(x.foo); // 正常工作，x被推断为Omit<A, 'baz'>
  }
}

原理说明 #

T extends unknown会触发分布式条件类型：当T是联合类型（比如A | B）时，TypeScript会自动将其拆分为A和B两个单独的类型，分别执行Omit<A, 'baz'>和Omit<B, 'baz'>，最后将结果重新组合为联合类型Omit<A, 'baz'> | Omit<B, 'baz'>，这样类型收窄就能正确识别x.type === 'A'对应的具体类型。

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