TypeScript：泛型函数无法正确推断约束泛型参数的问题

问题概述 #

我需要实现一个函数，接收返回Data接口对象的函数作为参数，同时保证Data接口的类型约束生效：subset必须是attributes指定的属性子集。

原接口定义 #

interface Data<TAttribute extends string = string> {
    readonly attributes: readonly TAttribute[]
    readonly subset: TAttribute extends infer U ? readonly U[] : never
}

正常场景：直接传入对象 #

当直接传入Data对象时，类型约束正常工作：

function dataNoFunc<TAttribute extends string>(data: Data<TAttribute>): Data<TAttribute> {
    return data
}

const goodNoFunc = dataNoFunc({ attributes: ['a', 'b'], subset: ['a']})
// 预期错误：`subset`传入`d`时触发类型检查失败
const badNoFunc = dataNoFunc({ attributes: ['a', 'b'], subset: ['d']})

问题场景：传入返回对象的函数 #

但当参数是返回Data对象的函数时，类型约束失效，TAttribute被推断为宽泛的string类型，导致subset可以传入任意字符串：

function dataWithFunc<const TFunction extends (() => Data)>(dataFunction: TFunction): TFunction {
    return dataFunction
}

const goodWithFunc = dataWithFunc(() => ({ attributes: ['a', 'b'], subset: ['a']}))
// 未触发预期错误：`subset`传入`d`仍通过检查
const badWithFunc = dataWithFunc(() => ({ attributes: ['a', 'b'], subset: ['d']}))

尝试泛化的失败尝试 #

我尝试让函数本身泛化，但出现子类型匹配错误：

function dataWithFuncAndGeneric<const TFunction extends (<const T extends string, R>() => Data<T>)>(dataFunction: TFunction): TFunction {
    return dataFunction
}

// 类型错误：子类型不匹配
const goodWithFuncAndGeneric = dataWithFuncAndGeneric(() => ({ attributes: ['a', 'b'], subset: ['a']} as const))
// 约束仍失效
const badWithFuncAndGeneric = dataWithFuncAndGeneric(() => ({ attributes: ['a', 'b'], subset: ['d']}))

错误提示：Type 'string' is not assignable to type 'T'. 'string' is assignable to the constraint of type 'T', but 'T' could be instantiated with a different subtype of constraint 'string'.

作为库代码，我希望无需用户手动指定泛型或添加断言即可解决问题。

解决方案 #

方案1：显式绑定泛型并使用const断言 #

直接将泛型参数绑定到Data的属性类型，同时要求attributes使用const断言保留字面量类型：

function dataWithFunc<const T extends string>(dataFunction: () => Data<T>): () => Data<T> {
    return dataFunction
}

// 正常工作
const goodWithFunc = dataWithFunc(() => ({ attributes: ['a', 'b'] as const, subset: ['a']}))
// 触发预期错误：`'d'`不在属性集合中
const badWithFunc = dataWithFunc(() => ({ attributes: ['a', 'b'] as const, subset: ['d']}))

方案2：自动推断字面量类型（无需手动断言） #

通过条件类型自动提取返回对象中attributes的字面量类型，完全无需用户额外操作：

function dataWithFunc<const T extends { attributes: readonly string[], subset: readonly string[] }>(
    dataFunction: () => T extends Data<infer U> ? Data<U> : never
): () => T extends Data<infer U> ? Data<U> : never {
    return dataFunction as () => T extends Data<infer U> ? Data<U> : never
}

// 自动推断字面量类型，无需`as const`
const goodWithFunc = dataWithFunc(() => ({ attributes: ['a', 'b'], subset: ['a']}))
// 触发预期错误
const badWithFunc = dataWithFunc(() => ({ attributes: ['a', 'b'], subset: ['d']}))

核心原理 #

• 让泛型参数直接关联Data的属性类型，避免TypeScript推断为宽泛的string
• 利用条件类型或const断言保留字面量类型，确保subset的约束基于具体的属性集合
• 简化函数的泛型约束逻辑，让类型推断自动完成

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