TypeScript为何无法根据参数推断正确返回类型？如何解决？

为什么无法识别参数对应的返回类型？ #

当前函数的参数arg是SetResponseType联合类型，但TypeScript默认不会建立参数具体取值和返回类型的绑定关系。它只会推断出函数的返回类型是number | string——因为知道函数可能返回这两种类型，但不会自动追踪参数分支对返回值的影响，无法根据传入的具体参数值缩小返回类型的范围。

实现参数与返回类型关联的方法 #

方法一：使用函数重载 #

通过定义多个重载签名，明确指定不同参数值对应的返回类型，TypeScript就能匹配对应的重载并推断正确返回类型：

type SetResponseType = 'returnNumber' | 'returnString';

// 重载签名：定义参数与返回类型的对应关系
function argumentTypedFunction(arg: 'returnNumber'): number;
function argumentTypedFunction(arg: 'returnString'): string;

// 实现签名：实际函数逻辑
function argumentTypedFunction(arg: SetResponseType): number | string {
    if (arg === 'returnNumber') {
        return 123;
    }
    return 'abc';
}

argumentTypedFunction('returnNumber'); // 类型为number
argumentTypedFunction('returnString'); // 类型为string

方法二：泛型结合条件类型 #

用泛型捕获传入的具体参数类型，再通过条件类型建立参数到返回类型的映射：

type SetResponseType = 'returnNumber' | 'returnString';

// 条件类型：定义参数类型到返回类型的映射
type ReturnTypeMap<T extends SetResponseType> = 
    T extends 'returnNumber' ? number : string;

const argumentTypedFunction = <T extends SetResponseType>(arg: T): ReturnTypeMap<T> => {
    if (arg === 'returnNumber') {
        return 123 as ReturnTypeMap<T>;
    }
    return 'abc' as ReturnTypeMap<T>;
}

argumentTypedFunction('returnNumber'); // 类型为number
argumentTypedFunction('returnString'); // 类型为string

这里的类型断言是辅助TypeScript确认分支返回值符合条件类型的要求，避免编译报错。

方法三：索引类型映射 #

先定义参数与返回类型的映射对象，再通过索引类型直接关联：

// 映射对象：明确参数值对应的返回类型
type ResponseMap = {
    returnNumber: number;
    returnString: string;
}

const argumentTypedFunction = <T extends keyof ResponseMap>(arg: T): ResponseMap[T] => {
    const responses: ResponseMap = {
        returnNumber: 123,
        returnString: 'abc'
    };
    return responses[arg];
}

argumentTypedFunction('returnNumber'); // 类型为number
argumentTypedFunction('returnString'); // 类型为string

这种方式更直观，不需要额外的类型断言，逻辑也更清晰。

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