映射类型场景下泛型函数返回值无法赋值的问题排查

问题代码 #

type Variants0 = {
  variant0: { v0: string };
  variant1: { v1: string };
};

const callAllVariants = <S extends { [P in keyof Variants0]:any }>(
  f: <V extends keyof S>(m: V) => S[V],
): { [V in keyof Variants0]: ReturnType<typeof f<V>> } => {
  return {
    variant0: f('variant0'),
    variant1: f('variant1'),
  };
};

const r0 = callAllVariants<Variants0>(<V extends keyof Variants0>(v: V) => {
  return {
    variant0: { v0: 'v0' },
    variant1: { v1: 'v1' },
  }[v];
});

type Variants1 = { [V in keyof Variants0]: Variants0[V] & { extra: number } };

const r1 = callAllVariants<Variants1>(
  <V extends keyof Variants1>(vs: V): Variants1[V] => {
    const r = {
      variant0: { v0: 'v0', extra: 0 },
      variant1: { v1: 'v1', extra: 1 },
    }[vs];
    // 无法正常编译：
    const r3: Variants0[V] & { extra: number } = r;
    const r2: Variants1[V] = r3;
    return r2;
    // 正常编译：
    // return r;
  },
);

编译错误信息 #

Type 'Variants0[V] & { extra: number; }' is not assignable to type 'Variants1[V]'.
  Type 'Variants0[V] & { extra: number; }' is not assignable to type '{ v0: string; } & { extra: number; } & { v1: string; }'.
    Type 'Variants0[V] & { extra: number; }' is not assignable to type '{ v0: string; }'

问题原因 #

核心问题是TypeScript对泛型条件下的类型推断存在限制：

• Variants1[V]是通过映射类型定义的，当V被约束为keyof Variants1时，TypeScript会默认将V视为可能是联合类型（即"variant0" | "variant1"），而非单个具体键。
• 你写的Variants0[V] & { extra: number }会被展开为(Variants0["variant0"] | Variants0["variant1"]) & { extra: number }，也就是({ v0: string } | { v1: string }) & { extra: number }，等价于({ v0: string } & { extra: number }) | ({ v1: string } & { extra: number })。
• 但TypeScript在处理泛型交叉推导时，错误地将Variants1[V]推断为Variants0["variant0"] & Variants0["variant1"] & { extra: number }（两个变体类型的交叉），而非对应键的具体交叉类型，导致类型不匹配。

直接返回r时编译正常，是因为TypeScript可以直接从对象字面量推断出r的类型是Variants1[V]的具体实例，跳过了中间泛型交叉类型的推导环节。

解决方法 #

如果必须保留中间类型，可通过以下两种方式解决：

1. 使用类型断言跳过检查：

const r2: Variants1[V] = r3 as Variants1[V];

2. 利用字面量类型的推断缩小范围：

const r1 = callAllVariants<Variants1>(
  <V extends keyof Variants1>(vs: V): Variants1[V] => {
    const r = {
      variant0: { v0: 'v0', extra: 0 },
      variant1: { v1: 'v1', extra: 1 },
    }[vs];
    const r3: Variants0[V] & { extra: number } = r;
    const r2 = r3 as typeof r;
    return r2;
  },
);

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