如何为添加对象字段的函数实现精确的TypeScript类型定义？

问题根源在于你没有让TypeScript捕获具体的字段名字面量类型，原代码中fieldName被定义为宽泛的string类型，导致类型系统只能推断出任意字符串索引，而非你传入的具体字段名。要实现预期的类型推导，需要为字段名新增泛型参数来锁定精确键名。

修改后的代码如下：

function addComputedField<T extends Record<string, unknown>, K extends string, V>(
  document: T,
  fieldName: K,
  computation: () => V
): T & { [key in K]: V };
function addComputedField<T extends Record<string, unknown>, K extends string, V>(
  document: T[],
  fieldName: K,
  computation: (doc: T) => V
): (T & { [key in K]: V })[];

function addComputedField<T extends Record<string, unknown>, K extends string, V>(
  document: T | T[],
  fieldName: K,
  computation: (doc?: T) => V
) {
  if (Array.isArray(document)) {
    return document.map((doc) => ({
      ...doc,
      [fieldName]: computation(doc),
    })) as (T & { [key in K]: V })[];
  } else {
    return { ...document, [fieldName]: computation() } as T & { [key in K]: V };
  }
}

核心修改说明 #

• 新增泛型参数K extends string：用来捕获调用时传入的fieldName的具体字面量类型（比如示例中的"d"）
• 调整返回类型：将原有的{ [field: typeof fieldName]: V }改为{ [key in K]: V }，让TypeScript把K解析为具体字段名，而非任意字符串
• 类型断言补充：实现部分添加类型断言，帮助TypeScript确认map和对象扩展后的精确类型

测试效果 #

调用示例代码：

const obj = { a: 1, b: 2, c: 3 }
const withComputed = addComputedField(obj, "d", () => 4)

此时withComputed的类型会被正确推导为：

const withComputed: {
    a: number;
    b: number;
    c: number;
} & {
    d: number;
}

在TypeScript 4.9及以上版本中，交叉类型会自动合并为扁平对象显示，最终类型会直接呈现为：

const withComputed: {
    a: number;
    b: number;
    c: number;
    d: number;
}

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