TypeScript中如何根据输入字段指定数据库查询函数的返回类型？

核心需求 #

实现数据库查询函数，让传入的字段列表直接决定返回对象的键类型，同时支持编译时已知字段和运行时动态字段两种场景。

正确的类型实现 #

1. 编译时确定字段（强类型约束） #

利用 TypeScript 泛型和 Pick 工具类型，让函数自动推导返回类型与输入字段的关联：

type MyTable = {
  id: string,
  name: string,
  age: number
};

async function QueryDatabaseTable<T extends keyof MyTable>(
  fieldsToReturn: T[]
): Promise<Pick<MyTable, T>[]> {
  // 实际调用时需确保字段数组正确传递给数据库驱动，这里简化示例
  const result = await dbClient.query('myTable').select(fieldsToReturn as string[]);
  return result;
}

调用示例（编译时类型推导）

// 传入具体字段数组，TypeScript 自动推导返回类型
const data = await QueryDatabaseTable(['name', 'age']);
// data 的类型为 { name: string; age: number; }[]
// 可直接访问 name/age，访问 id 会触发 TypeScript 报错（符合预期）
console.log(data[0].name, data[0].age);

2. 运行时动态字段（宽松类型约束） #

如果字段是运行时才确定的（比如前端用户选择的报表字段），TypeScript 无法捕获具体键类型，但可以约束返回值为 MyTable 的部分属性：

async function QueryDatabaseTableDynamic(
  fieldsToReturn: (keyof MyTable)[]
): Promise<Partial<MyTable>[]> {
  const result = await dbClient.query('myTable').select(fieldsToReturn as string[]);
  return result;
}

调用示例（运行时动态字段）

// 模拟用户动态选择的字段
const userSelectedFields: (keyof MyTable)[] = ['id', 'name'];
const dynamicData = await QueryDatabaseTableDynamic(userSelectedFields);
// dynamicData 的类型为 Partial<MyTable>[]，访问属性时需做可选链判断
console.log(dynamicData[0]?.name);

原有方案的问题分析 #

• 最初实现中，泛型 T 被设为 keyof MyTable，但参数是单键而非键数组，且返回类型为键的数组而非对象数组，完全不符合需求。
• 备选方案的 KeysToArrays 类型错误地将参数定义为「每个键对应数组的对象」，而非实际需要的「键的数组」，同时 Partial<MyTable>[] 无法关联输入字段和返回类型，丢失了强类型约束。

关键知识点 #

• Pick<T, K>：TypeScript 内置工具类型，从类型 T 中提取指定键 K，生成新的类型。
• 泛型约束 T extends keyof MyTable：确保传入的字段只能是 MyTable 中存在的键，避免非法字段输入。
• 类型断言 as string[]：部分数据库驱动仅接受字符串数组，用断言解决类型兼容问题（若驱动支持泛型则无需断言）。

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