如何自动推导预定义对象的Key与Value类型？

问题场景 #

我有一个映射对象和根据值获取键的辅助函数，代码如下：

const map = {
    0: 'a',
    1: 'b',
    2: 'c',
    3: 'd',
}

function getKey(value: Value): Key {
    return Object.keys(map).find(key => map[key] === value);
}

之前需要手动定义Key和Value类型：

type Key = 0 | 1 | 2 | 3 | '0' | '1' | '2' | '3'
type Value = 'a' | 'b' | 'c' | 'd'

现在想避免手动定义——因为对象可能变化甚至是函数返回结果，已知键是数字类型、值是字符串类型，但不想定义得太泛化，希望用泛型实现自动推导。

解决方案 #

可以利用TypeScript的内置类型工具和as const断言实现自动推导，具体步骤如下：

1. 锁定静态映射的字面量类型 #

如果映射是静态定义的，给对象加上as const断言，让TypeScript保留精确的键值字面量类型，而非宽泛的{ [key: number]: string }：

const map = {
    0: 'a',
    1: 'b',
    2: 'c',
    3: 'd',
} as const;

2. 自动推导Key和Value类型 #

通过typeof和keyof等内置工具，直接从映射对象推导精确类型：

// 推导原始数字键的字面量类型
type NumericKey = keyof typeof map;
// 推导值的字面量类型
type Value = typeof map[NumericKey];
// 兼容Object.keys返回的字符串键，同时保留原始数字键类型
type Key = NumericKey | `${NumericKey}`;

3. 修正getKey函数的类型匹配 #

由于Object.keys()返回的是string[]，直接返回会有类型不匹配问题，需要做类型断言，同时处理找不到值的边界情况：

function getKey(value: Value): Key | undefined {
    return Object.keys(map).find(key => map[Number(key) as NumericKey] === value) as Key | undefined;
}

4. 处理动态生成的映射对象 #

如果映射是函数返回的动态结果，用泛型捕获其具体类型：

function createMap() {
    return {
        0: 'a',
        1: 'b',
        2: 'c',
        3: 'd',
    } as const;
}

// 捕获函数返回值的精确类型
type MapType = ReturnType<typeof createMap>;
type NumericKey = keyof MapType;
type Value = MapType[NumericKey];
type Key = NumericKey | `${NumericKey}`;

const map = createMap();

function getKey(value: Value): Key | undefined {
    return Object.keys(map).find(key => map[Number(key) as NumericKey] === value) as Key | undefined;
}

后续修改映射对象时，Key和Value类型会自动同步更新，无需手动维护。

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