为何TypeScript中Map的取值类型无法被正确推断？

先看你遇到的场景：

const map1 = new Map([
  ["a", 1]
])
const a = map1.get("a") // 实际类型：number | undefined （期望：number）

const map2 = new Map()
map2.set("b", 2)
const b = map2.get("b") // 实际类型：any （期望：number）

为什么会这样？ #

• map1的情况：TS能根据构造参数推断出这是Map<string, number>，但Map.get()方法的设计本身就会返回value | undefined——毕竟TS无法在编译期100%确认你传入的键一定存在于Map里，哪怕你自己知道它存在，TS也会做安全校验。
• map2的情况：创建空Map时，TS没有任何类型线索，默认会把它推断成Map<any, any>，所以后续get()返回的自然就是any类型。

解决办法 #

1. 显式指定Map的泛型类型 #

直接声明键值的类型，同时用非空断言告诉TS你确定键存在（注意：非空断言要确保运行时键真的存在，否则会报错）：

// 处理map1
const map1 = new Map<string, number>([["a", 1]])
const a = map1.get("a")! // 类型变为number

// 处理map2
const map2 = new Map<string, number>()
map2.set("b", 2)
const b = map2.get("b")! // 类型变为number

2. 用类型守卫做安全检查 #

如果不想用非空断言（避免潜在的运行时风险），可以先判断值是否存在，TS会自动缩小类型：

const map1 = new Map<string, number>([["a", 1]])
const aValue = map1.get("a")
if (aValue !== undefined) {
  // 这里aValue的类型是number
  console.log(aValue.toFixed(2))
}

const map2 = new Map<string, number>()
map2.set("b", 2)
const bValue = map2.get("b")
if (bValue !== undefined) {
  // 此处bValue类型为number
  console.log(bValue.toFixed(2))
}

3. 封装一个严格类型的Map子类 #

如果经常需要这种“键一定存在”的场景，可以自己写个子类重写get方法，确保返回值一定是目标类型（不存在则抛出错误）：

class StrictMap<K, V> extends Map<K, V> {
  get(key: K): V {
    const value = super.get(key)
    if (value === undefined) {
      throw new Error(`找不到键：${String(key)}`)
    }
    return value
  }
}

const map1 = new StrictMap<string, number>([["a", 1]])
const a = map1.get("a") // 类型为number

const map2 = new StrictMap<string, number>()
map2.set("b", 2)
const b = map2.get("b") // 类型为number

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