TypeScript联合类型变量在闭包中的类型推断异常问题问询

以下是简化后的TypeScript代码：

function withHandler() { // 返回类型: number
    let a: string | number = 1
    handler(() => {
        a = 'string value'
    })
    return a
}

function noHandler() { // 返回类型: string
    let a: string | number = 1
    a = 'string value'
    return a
}

function handler(cb: () => void) {
    cb()
}

我预期withHandler()和noHandler()的隐式返回类型均为string | number，因为变量a是通过let声明的string | number联合类型。但实际withHandler()被推断为返回number（显然它会返回字符串），noHandler()被推断为返回string。请问该行为是否符合预期，还是TypeScript的bug？

解答 #

这是TypeScript的预期行为，并非bug，核心原因在于TypeScript的控制流分析范围和对void类型回调的处理逻辑：

1. noHandler()的类型推断逻辑
   在noHandler函数内部，变量a的赋值操作直接处于当前函数的控制流中，TypeScript的控制流分析可以完整追踪到a从初始number被重新赋值为string的过程，因此能准确推断出返回值类型为string。

2. withHandler()的类型推断逻辑
   handler函数的参数类型是() => void，TypeScript默认认为标注为void的回调函数不会产生修改外部变量的副作用，因此不会跨函数追踪这类回调内部对外部变量的修改。编译器看不到回调里对a的赋值操作，依然认为a保持初始的number类型，所以推断返回值为number。

如果需要让withHandler()的返回类型被正确推断，可以调整handler的类型定义，比如将回调参数类型改为更明确的类型，或者通过类型断言、显式指定返回类型来修正。

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