如何在TypeScript中定义键值严格映射的联合Map类型？

要实现键与值严格一一对应的Map，你需要先明确键值的固定映射关系，再基于这个关系自定义类型来约束Map的set和get方法。以下是具体方案：

1. 定义键值映射关系 #

首先用类型别名明确每个键对应的具体值类型（先假设你的基础类型如下）：

type KeyTypeA = 'keyA';
type ValueTypeA = number;
type KeyTypeB = 'keyB';
type ValueTypeB = string;

// 核心：定义键值的严格对应关系
type KeyValueMapping = {
  [K in KeyTypeA]: ValueTypeA;
  [K in KeyTypeB]: ValueTypeB;
};

2. 自定义强类型Map类型 #

基于上面的映射关系，创建一个覆盖set和get方法的自定义Map类型，同时保留原生Map的其他方法：

type TypedMap<T extends Record<PropertyKey, unknown>> = {
  // set方法：键必须是T的key，值必须匹配对应类型
  set<K extends keyof T>(key: K, value: T[K]): void;
  // get方法：返回对应键的具体值类型（加上undefined，因为Map可能无该键）
  get<K extends keyof T>(key: K): T[K] | undefined;
} & Omit<Map<keyof T, T[keyof T]>, 'set' | 'get'>;

3. 创建并使用强类型Map #

可以通过工厂函数或类型断言创建实例，工厂函数更优雅：

工厂函数实现 #

function createTypedMap<T extends Record<PropertyKey, unknown>>(): TypedMap<T> {
  return new Map() as TypedMap<T>;
}

const myBigMap = createTypedMap<KeyValueMapping>();

现在测试你的需求：

const valueA: ValueTypeA = 123;
const valueB: ValueTypeB = 'hello';

myBigMap.set('keyA', valueA); // ✅ 符合预期
myBigMap.set('keyA', valueB); // ❌ 报错：类型string不能赋值给number
const myValue = myBigMap.get('keyA'); // 🎯 类型为ValueTypeA | undefined

为什么之前的尝试无效？ #

• 普通Map<CombinedKeyType, CombinedValueType>：TypeScript无法追踪单个键对应的具体值类型，仅能识别键和值都是联合类型，因此无法限制键值对应关系，get返回值也是联合类型。
• 联合Map<KeyTypeA, ValueTypeA> | Map<KeyTypeB, ValueTypeB>：TypeScript会取两个set方法参数的交集，若KeyTypeA和KeyTypeB无重叠，参数类型会缩减为never，导致无法调用set。
• 交叉Map<KeyTypeA, ValueTypeA> & Map<KeyTypeB, ValueTypeB>：set方法会合并两个签名，能约束键值对应关系，但get方法的返回值会是ValueTypeA | ValueTypeB——因为TypeScript无法确定你获取的键属于交叉类型的哪一部分，只能返回联合类型。

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