TypeScript中如何捕获可选参数类型以推导函数返回值？

问题核心 #

原函数b的泛型约束仅限定为string，而可选参数b?: A的实际类型是A | undefined，但TypeScript不会自动将undefined纳入泛型A的推导范围。当调用b()时，泛型A会被推导为string，导致Ret<A>计算为2，但函数实际返回1，引发类型不匹配错误。

解决方案1：扩展泛型约束范围 #

修改泛型约束为string | undefined，让TypeScript在不传参数时能将A推导为undefined，从而正确匹配返回值类型：

type Ret<A> = A extends undefined ? 1 : 2

function b<A extends string | undefined>(b?: A): Ret<A> {
    if (b === undefined) {
        return 1 as Ret<A>
    } else {
        return 2 as Ret<A>
    }
}

// 调用验证
const resultUndefined2: 1 = b() // 类型匹配正确
const resultString2: 2 = b("abc") // 类型匹配正确

解决方案2：使用函数重载 #

通过函数重载明确不同调用场景的返回值类型，直接定义入参与返回值的对应关系，可读性更强：

type Ret<A> = A extends undefined ? 1 : 2

// 重载签名：无参数时返回1
function b(): 1;
// 重载签名：传入string时返回2
function b<A extends string>(b: A): 2;
// 函数实现
function b(b?: string): 1 | 2 {
    if (b === undefined) {
        return 1
    } else {
        return 2
    }
}

// 调用验证
const resultUndefined2: 1 = b() // 类型匹配正确
const resultString2: 2 = b("abc") // 类型匹配正确

说明 #

• 方案1通过扩展泛型约束，让TypeScript自动推导undefined场景的类型，保持代码简洁；
• 方案2通过重载直接定义类型规则，适合需要明确区分多种入参场景的复杂情况。

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