TypeScript带条件泛型的函数参数类型推断问题求助

问题代码 #

enum TYPES {
    ME = 'me',
    YOU = 'you'
}

type Attributes<T extends TYPES> =  T extends TYPES.ME ? {keys: true} : {hat: true}
const func = <T extends TYPES>(type: T, attr: Attributes<T>) => {
    if(type === TYPES.ME) {
        attr.keys // 报错：Property 'keys' does not exist on type '{ keys: true; } | { hat: true; }'
    }
} 

问题原因 #

TypeScript无法在泛型函数内部通过枚举值判断，自动关联缩小attr的类型。因为泛型T是TYPES的子集（可能是单个枚举值或整个联合类型），当判断type === TYPES.ME时，编译器没法确定此时T就等于TYPES.ME，因此attr的类型仍被视为Attributes<TYPES.ME> | Attributes<TYPES.YOU>的联合类型，无法直接访问keys属性。

解决方案 #

方案1：使用函数重载（类型安全，保留原有参数结构）

通过重载签名明确每个枚举值对应的属性类型，让编译器能精准推断：

enum TYPES {
    ME = 'me',
    YOU = 'you'
}

type Attributes<T extends TYPES> = T extends TYPES.ME ? {keys: true} : {hat: true}

// 重载签名：明确枚举值与属性的对应关系
function func(type: TYPES.ME, attr: Attributes<TYPES.ME>): void;
function func(type: TYPES.YOU, attr: Attributes<TYPES.YOU>): void;
// 实现签名：处理逻辑
function func(type: TYPES, attr: Attributes<TYPES>) {
    if(type === TYPES.ME) {
        attr.keys; // 无报错，类型推断正确
    } else {
        attr.hat; // 同样能正确推断
    }
}

这种方式既保留了原有的参数传递方式，又能在函数内部获得准确的类型推断，同时调用时会严格检查参数匹配。

方案2：使用判别式联合参数（更直观的类型关联）

将type和attr合并为一个包含判别式的联合类型对象，TypeScript会自动根据判别式缩小类型范围：

enum TYPES {
    ME = 'me',
    YOU = 'you'
}

// 定义判别式联合类型，用type作为判别字段
type FuncParams = 
    | { type: TYPES.ME; attr: {keys: true} }
    | { type: TYPES.YOU; attr: {hat: true} };

const func = (params: FuncParams) => {
    if(params.type === TYPES.ME) {
        params.attr.keys; // 自动推断为{keys: true}类型
    } else {
        params.attr.hat; // 自动推断为{hat: true}类型
    }
}

这种方案的优势是类型关联更直观，编译器无需额外泛型推导，就能完美识别类型分支，适合参数关联性强的场景。

方案3：临时解决方案——类型断言（不推荐长期使用）

如果只是临时快速解决，可以手动将attr断言为对应类型，但这种方式会跳过类型检查，存在安全风险：

const func = <T extends TYPES>(type: T, attr: Attributes<T>) => {
    if(type === TYPES.ME) {
        (attr as Attributes<TYPES.ME>).keys; // 强制断言，无报错
    }
}

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