TypeScript 实现对象键下Datum数组泛型的精准类型推断

当前泛型声明将TKey作为独立前置泛型参数，TS做类型推断时，会按照找最宽泛满足约束值的逻辑，将所有符合Datum<TKey>[]约束的可能值合并为最大联合类型string | number，完全不会根据传入的key参数逐属性匹配对应Datum的泛型实参，最终导致返回值类型退化为宽泛的联合类型数组。
原泛型的映射约束T extends { [k in K]: Datum<TKey>[] }本质要求对象下所有匹配K的属性值，都必须是同一个泛型参数TKey下的Datum数组，和“不同属性存储不同泛型类型Datum数组”的实际业务场景完全不匹配。

完全去掉前置声明的TKey泛型参数，借助TS的索引访问类型直接取对应键的源数组类型即可，代码实现如下：

export interface Datum<T extends string | number> {
    key: T;
    count: number;
}

const aggregateDatums = <
    T extends Record<keyof any, Datum<string | number>[]>,
    K extends keyof T
>(data: T, key: K): T[K] => {
    // 原有业务逻辑保持不变
    return data[key];
};

效果验证：

const data = {
    numeric: [{key: 1, count: 1}, {key: 2, count: 2}],
    text: [{key: 'a', count: 1}, {key: 'b', count: 2}],
};

const aggregated = {
    numeric: aggregateDatums(data, 'numeric'), // 自动推断为 Datum<number>[]
    text: aggregateDatums(data, 'text'), // 自动推断为 Datum<string>[]
};

这里的入参对象约束要求每个属性的值都必须是Datum数组，和业务场景完全对齐，TS会在实际调用时保留每个属性下Datum的具体泛型实参（number或string），不会丢失类型信息。

• Datum接口本身已经声明了泛型参数T extends string | number，不需要在函数层面重复声明TKey做相同约束
• 独立前置的TKey泛型参数会打断TS的逐属性类型推断链路，让TS无法拆分保留不同属性下的Datum泛型类型，只能合并成宽泛的联合类型
• 整体泛型参数数量比实际需要多了一层，没有必要的约束反而引入了类型推断的bug。

• 基础阶段：不要只停留在会用Partial<T>、Pick<T,K>这类内置工具类型，逐行吃透内置工具类型的源码实现，掌握映射类型、条件类型、infer推导、索引访问类型的核心用法
• 实践阶段：从业务场景出发写小型类型工具，比如“提取对象中值为函数的属性名”“打平嵌套数组类型”这类小需求，不要一开始就啃复杂度极高的类型体操题目
• 进阶阶段：熟悉TS的核心推断规则，尤其是泛型参数的推断优先级、协变/逆变对推断结果的影响——本次遇到的类型退化问题，本质是未掌握TS对独立泛型参数“取最大满足约束范围”的推断逻辑。

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