如何让TypeScript泛型函数test仅接受含b可选属性的类型参数？

我需要创建一个泛型函数test，它接受类型为T的参数并返回T类型的值。T需要满足：是包含必选属性a（string类型）、显式包含可选属性b（number | undefined类型）的对象。

当前代码的问题是，对于不包含b属性的类型C，调用test(c)时没有触发类型错误，但实际函数内部会给arg.b赋值123，导致后续用zod strict模式解析时失败（因为C不允许存在b属性）。

解决方案 #

要解决这个问题，需要修改泛型约束，强制T必须显式包含b属性（即使是可选的）。可以通过交叉类型结合条件检查，确保'b'是T的键之一：

function test<T extends { a: string, b?: number } & ('b' extends keyof T ? unknown : never)>(arg: T) {
  arg.b = 123;
  return arg;
}

代码验证 #

修改后的代码会对以下场景做出正确反应：

1. 类型A显式包含b?: number，正常通过：

type A = { a: string, b?: number };
const a: A = { a: "1" };
test(a); // 无错误

2. 类型B在包含额外属性的同时显式声明b?: number，正常通过：

type B = { a: string, b?: number, c: string };
const b: B = { a: "1", c: "hello" };
test(b); // 无错误

3. 类型C未声明b属性，触发类型错误（符合预期）：

type C = { a: string, c: string };
const c: C = { a: "1", c: "hello" };
test(c); // 报错：类型“C”的参数不能赋给类型“{ a: string; b?: number | undefined; } & ('b' extends keyof C ? unknown : never)”的参数。

原理说明 #

• 原约束T extends { a: string, b?: number }允许T省略b属性，因为TypeScript中可选属性的语义是“存在或不存在均可”。
• 添加的& ('b' extends keyof T ? unknown : never)约束会在T不包含b作为键时，将交叉类型变为never，导致T无法满足泛型约束，从而触发类型错误。
• 这样就能保证传入test的参数类型都显式声明了b属性，后续给arg.b赋值时不会产生超出类型预期的属性，避免zod strict解析失败。

内容的提问来源于stack exchange，提问作者Page not found