TypeScript泛型函数中枚举映射类型赋值失败问题求助

原因分析 #

报错的核心是泛型约束与映射类型的索引访问兼容性限制：

• 泛型约束T extends Foo允许T为Foo的联合类型（如Foo.A | Foo.B），而非仅单个枚举成员。
• 当T为泛型时，AllFoo[T]会被解析为所有FooType<K> | undefined的联合（K遍历Foo所有成员）；而FooType<T>是对应T的具体类型，TypeScript无法自动推导两者的子集关系，因此判定FooType<T>无法赋值给AllFoo[T]——尽管从逻辑上讲，可选属性AllFoo[T]本应接受FooType<T>类型的值。
• 本质是TypeScript类型系统在处理泛型与同态映射类型的交互时，缺乏对“键与类型严格绑定”的自动推断能力。

无断言解决方案 #

最优方案：为函数参数定义泛型专属映射类型 #

修改函数的all参数类型，使其严格对应当前泛型T的键值关系，让TypeScript明确索引访问的类型兼容性：

enum Foo {A, B, C}

interface FooData {
    [Foo.A]: null,
    [Foo.B]: {x: number, y: string},
    [Foo.C]: {x: boolean, z: number[]},
}

interface FooType<T extends Foo> {
    type: T;
    data: FooData[T];
}
type AllFoo = {[k in Foo]?: FooType<k>}

function setFooData<T extends Foo>(all: { [K in T]: FooType<K> | undefined }, type: T, data: FooData[T]) {
    all[type] = {type: type, data: data};
}

function setFooType<T extends Foo>(all: { [K in T]: FooType<K> | undefined }, val: FooType<T>) {
    all[val.type] = val;
}

const allFoo: AllFoo = {};
setFooData(allFoo, Foo.B, {x: 1, y: "2"});
setFooType(allFoo, {type: Foo.C, data: {x: true, z: [1,2,3]}});

console.log(`y is ${allFoo[Foo.B]?.data.y}`);

该方案通过为泛型T生成专属映射类型，让TypeScript直接识别all[type]的类型为FooType<T> | undefined，而赋值的{type, data}正好匹配FooType<T>，完全符合赋值规则，无需任何类型断言。

备选方案：收窄泛型至单个枚举成员 #

若需保留AllFoo作为参数类型，可通过Extract<Foo, T>强制T为单个枚举成员，结合类型收窄实现（但此方案仍需一处类型断言，仅作参考）：

function setFooData<T extends Foo>(all: AllFoo, type: T, data: FooData[Extract<Foo, T>]) {
    const fooType: FooType<Extract<Foo, T>> = {type: type as Extract<Foo, T>, data};
    all[type] = fooType;
}

验证 #

修改后的代码在TypeScript 4.6.4及5.0.4版本中均可正常编译，且完整保留类型安全特性。

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