TypeScript类型安全Pipe函数实现中的类型推断问题

问题原因分析 #

第一个简化示例的问题 #

你定义的pipe函数泛型T被推断为string（因为输入数组是string[]），此时fn的类型是((a: string[]) => string) | string。而first函数的类型是<T extends any[]>(x: T) => T[0]，它的参数要求是任意数组类型，和pipe期望的string[]参数无法自动对齐——TypeScript没办法把first的泛型T（代表任意数组）和pipe的泛型T（代表string）关联起来，最终导致类型推断失败，返回any。

第二个场景示例的问题 #

调整后的pipe泛型A被推断为string[]，fn1的类型是((a: [string[]]) => B) | string[]。直接传入first时，TypeScript无法自动推导出first的泛型T应该是string——因为B没有被函数返回值的类型约束，所以默认推断为unknown。而用箭头函数包裹时，箭头函数的参数x被明确推断为[string[]]，调用first(x)时T会被正确推断为string，所以返回类型正常。

解决方案 #

核心思路是明确区分pipe的两种入参情况（函数/非函数），避免联合类型导致的推断歧义，推荐用函数重载或条件类型来实现：

方案1：函数重载（更直观） #

针对第一个简化示例的修复：

// 重载1：传入转换函数
function pipe<T, R>(x: T[], fn: (a: T[]) => R): R;
// 重载2：传入直接返回值
function pipe<T>(x: T[], fn: T): T;
// 函数实现
function pipe<T, R>(x: T[], fn: ((a: T[]) => R) | T): R | T {
    if (typeof fn === 'function') return fn(x);
    return fn;
}

function first<T extends any[]>(x: T): T[0] {
    return x[0];
}

const result1 = pipe(["s", "t"], first);
//    ^? string（类型推断正常）
const result2 = pipe(["s", "t"], (x) => x[0]);
//    ^? string（类型推断正常）

针对第二个场景示例的修复：

// 重载1：传入转换函数
function pipe<A, R>(x: A, fn: (a: [A]) => R): R;
// 重载2：传入直接返回值
function pipe<A>(x: A, fn: A): A;
// 函数实现
function pipe<A, R>(x: A, fn: ((a: [A]) => R) | A): R | A {
    return typeof fn === 'function' ? fn([x]) : fn;
}

function first<T>(x: [T[]]): T {
    return x[0][0];
}

const result1 = pipe(["s", "t"], first);
//    ^? string（类型推断正常）
const result2 = pipe(["s", "t"], (x) => first(x));
//    ^? string（类型推断正常）

方案2：条件类型（更简洁） #

用条件类型直接约束返回值：

function pipe<A, F extends ((a: [A]) => unknown) | A>(
    x: A,
    fn: F
): F extends ((a: [A]) => infer R) ? R : A;

function pipe<A, F extends ((a: [A]) => unknown) | A>(
    x: A,
    fn: F
) {
    return typeof fn === 'function' ? fn([x]) : fn;
}

function first<T>(x: [T[]]): T {
    return x[0][0];
}

const result1 = pipe(["s", "t"], first);
//    ^? string
const result2 = pipe(["s", "t"], (x) => first(x));
//    ^? string

总结 #

原代码的问题在于联合类型模糊了TypeScript的推断方向，无法将传入的泛型函数（如first）的类型与pipe期望的参数/返回值类型建立明确关联。通过重载或条件类型明确区分入参场景，就能让TypeScript正确完成类型推断。

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