TypeScript中扩展动态对象并保留所有有效键联合类型的实现

要实现对象扩展后保留精确的键联合类型，核心是利用TypeScript的泛型推断和交叉类型，避免使用Record<string, number>这类宽泛的类型定义。

具体实现步骤 #

1. 初始对象类型定义
   不要用宽泛的泛型约束，让TypeScript自动推断初始对象的字面量类型，或者显式指定具体的键值类型：

   // 自动推断类型为 { a: number; b: number }
   let obj = { a: 1, b: 2 };
   // 或者显式指定类型
   // let obj: { a: number; b: number } = { a: 1, b: 2 };
   

2. 泛型扩展函数
   定义泛型函数，接收原有对象的类型T，返回T与新增键值对的交叉类型，这样TypeScript会保留原有键的精确类型：

   function addSomething<T extends object>(obj: T): T & { x: number } {
     return { ...obj, x: 3 };
   }
   

3. 重新赋值并验证类型
   调用扩展函数后重新赋值给obj，此时keyof typeof obj会自动推断为'a' | 'b' | 'x'：

   obj = addSomething(obj);
   
   // 验证：此时访问obj.a、obj.b、obj.x都有类型提示，keyof typeof obj为 'a'|'b'|'x'
   type ObjKeys = keyof typeof obj; // 类型为 'a' | 'b' | 'x'
   

更灵活的扩展方式 #

如果需要支持添加任意键值对，可以扩展泛型函数，允许传入自定义的扩展对象：

function extendObject<T extends object, U extends object>(obj: T, extension: U): T & U {
  return { ...obj, ...extension };
}

// 使用示例：添加x和y两个键
obj = extendObject(obj, { x: 3, y: 4 });
type ObjKeys = keyof typeof obj; // 类型为 'a' | 'b' | 'x' | 'y'

关键原理 #

• 避免使用Record<string, number>，因为它会把键类型宽泛化为string，丢失具体的键联合类型
• 泛型T会捕获传入对象的精确键类型，交叉类型T & U会合并原有键和新增键的类型，让TypeScript能正确推断出完整的键联合类型

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