如何为TypeScript中处理混合参数的转数组函数正确推断/声明类型？

我想要一个简单的工具函数，将值按需转为数组，初始代码如下：

function castToArray(args: unknown) {
  return Array.isArray(args) ? args : [args];
}

理想情况下，希望TypeScript能从args的类型推断返回数组的类型，但默认返回类型为any[]。

我尝试了泛型方案：

function castToArray<T>(args: T) {
   return Array.isArray(args) ? args : [args];
}

此时返回类型为(T & any[]) | T[]，但单个元素仍被视为any类型。

随后我尝试使用条件返回类型：

function castToArray<T>(args: T): T extends any[] ? T : T[] {
    return Array.isArray(args) ? args : [args];
}

该方案能正确推断返回类型，但TypeScript报出赋值错误：

Type '(T & any[]) | T[]' is not assignable to type 'T extends any[] ? T : T[]'.
Type 'T & any[]' is not assignable to type 'T extends any[] ? T : T[]'

请问我忽略了什么？

解答 #

这个错误的核心是：TypeScript的条件类型在函数实现阶段无法被编译器直接解析为具体分支，它没办法自动验证(T & any[]) | T[]是否严格匹配条件类型T extends any[] ? T : T[]的预期结果。

这里有两种可行的解决方式：

方案1：类型断言绕过检查 #

既然我们明确逻辑符合条件类型的预期，可以给返回值添加类型断言：

function castToArray<T>(args: T): T extends any[] ? T : T[] {
    return (Array.isArray(args) ? args : [args]) as T extends any[] ? T : T[];
}

方案2：函数重载（更推荐） #

函数重载可以清晰定义不同输入对应的输出类型，同时避免条件类型的实现冲突：

// 重载签名：输入数组时返回原数组类型
function castToArray<T extends any[]>(args: T): T;
// 重载签名：输入非数组时返回单元素数组类型
function castToArray<T>(args: T): T[];
// 实现签名：处理逻辑，类型可以宽松处理
function castToArray(args: unknown) {
    return Array.isArray(args) ? args : [args];
}

这种方式既能保证类型推断完全符合预期，代码可读性也更强，编译器不会再抛出错误。

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