如何实现TypeScript中Map的键自动补全功能？

要实现Map的键自动补全，核心是让TypeScript识别Map的键为具体的字符串字面量联合类型，而非宽泛的string类型。以下是几种无需手动定义Keys类型的实现方式：

1. 初始化时用const断言推导类型 #

直接在创建Map时传入键值对数组，并添加as const断言，TypeScript会自动推导键的联合类型：

const map = new Map([
  ["key1", 1],
  ["key2", "hello world"]
] as const);

map.get("key1"); // 此处会自动补全key1、key2

2. 用泛型工具函数约束键类型 #

如果需要动态调用set方法添加键，可通过泛型函数创建Map，后续的set操作会自动扩展键的联合类型：

function createMap<K extends string, V>(initialEntries?: [K, V][]) {
  return new Map<K, V>(initialEntries);
}

// 初始化传入第一个键值对，键类型初始为"key1"
const map = createMap([["key1", 1]]);
// 调用set添加新键，键类型自动扩展为"key1" | "key2"
map.set("key2", "test");

map.get("key"); // 输入时会自动补全key1、key2，错误键会触发类型提示

3. 从对象推导Map键类型 #

如果你的键值对原本存储在对象中，可以通过keyof推导键类型，再转为Map：

const keyValueObj = {
  key1: 100,
  key2: true
} as const;

// 自动推导键的联合类型
type MapKeys = keyof typeof keyValueObj;
const map = new Map<MapKeys, typeof keyValueObj[MapKeys]>(
  Object.entries(keyValueObj) as [MapKeys, typeof keyValueObj[MapKeys]][]
);

map.get("key1"); // 自动补全key1、key2

关于Record类型的说明 #

Record类型主要用于定义对象的键值类型，比如Record<"key1" | "key2", any>，它本身不能直接实现Map的自动补全，但可以结合对象转Map的方式间接使用，本质还是依赖keyof推导具体的键类型。

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