如何修改callbackWrapper函数泛型类型定义以消除返回类型错误

现有如下TypeScript代码：

function callbackWrapper<T = string>(cb?: () => T): T  {
    return cb?.() ?? 'default value';
}

const a = callbackWrapper(() => 7); // 期望 a 为 number 类型
const b = callbackWrapper(() => true); // 期望 b 为 boolean 类型
const c = callbackWrapper(() => 'string'); // 期望 c 为 string 类型
const d = callbackWrapper(); // 期望 d 为 string 类型（值为'default value'）

当前函数返回类型存在错误：当传入返回非string类型的回调时，函数实际可能返回T或string（默认值），但原定义的返回类型仅为T，导致类型不兼容；用类型断言绕开错误并非合理解法。

推荐方案：使用函数重载 #

函数重载能明确区分两种调用场景的返回类型，不用类型断言就能满足需求：

// 传回调时，返回回调的返回类型
function callbackWrapper<T>(cb: () => T): T;
// 不传回调时，返回string类型
function callbackWrapper(): string;
// 函数实现
function callbackWrapper<T = string>(cb?: () => T) {
    return cb?.() ?? 'default value';
}

为什么这样可行：

• 当你传入回调函数时，TypeScript会匹配第一个重载签名，自动推导返回类型为回调的返回值类型，比如callbackWrapper(() => 7)的返回类型就是number；
• 当你不传回调时，会匹配第二个重载签名，返回类型直接是string，和默认值的类型完全对应；
• 函数实现部分不需要额外加类型断言，TypeScript会根据重载的规则自动约束类型。

替代方案：调整泛型返回类型（需类型断言） #

如果不想用重载，也可以通过条件类型明确返回类型，但实现时需要临时加类型断言，不如重载直观：

function callbackWrapper<T = string>(cb?: () => T): T | (undefined extends typeof cb ? string : never) {
    return (cb?.() ?? 'default value') as T | string;
}

内容的提问来源于stack exchange，提问作者Žilvinas Vidmantas