如何在TypeScript中构建类型安全的数据处理流水线？

在大数据应用的复杂数据转换场景中（如过滤、映射、聚合），保证转换流水线的类型安全是避免运行时错误的关键。现有实现无法静态验证步骤间的类型衔接，我们可以利用TypeScript的可变元组类型和递归类型约束来解决这个问题。

现有实现的问题 #

当前的createPipeline函数使用any类型规避了类型检查，既无法保证前一步输出与后一步输入的类型匹配，返回值类型也错误地取了所有步骤返回类型的联合，而非最后一步的结果：

type Transformation<T, U> = (input: T) => U;

// 存在类型安全漏洞的实现
function createPipeline<T, Steps extends [...Transformation<any, any>[]]>(
  initialData: T,
  ...steps: Steps
): ReturnType<Steps[number]> {
  return steps.reduce((data, step) => step(data), initialData);
}

类型安全的实现方案 #

通过递归可变元组类型，我们可以强制要求每个转换步骤的输入类型严格等于前一步的输出类型，同时准确推导最终返回值类型：

type Transformation<T, U> = (input: T) => U;

// 递归约束转换步骤的类型链：每个步骤的输入必须是前一步的输出
type TransformationChain<Initial> = 
  | [Transformation<Initial, infer Next>]
  | [Transformation<Initial, infer Next>, ...TransformationChain<Next>];

function createPipeline<Initial, Chain extends TransformationChain<Initial>>(
  initialData: Initial,
  ...steps: Chain
): ReturnType<Chain[Chain['length'] - 1]> {
  return steps.reduce((data, step) => step(data), initialData) as ReturnType<Chain[Chain['length'] - 1]>;
}

示例验证 #

使用正确的步骤顺序时，TypeScript会正常推导类型：

const parseNumbers = (data: string[]) => data.map(Number);
const filterValidNumbers = (data: number[]) => data.filter(n => !isNaN(n));
const sumNumbers = (data: number[]) => data.reduce((sum, n) => sum + n, 0);

// 正确用法：步骤类型衔接无误，result类型被推导为number
const result = createPipeline(['1', '2', 'three'], parseNumbers, filterValidNumbers, sumNumbers);
console.log(result); // 输出：3

错误场景拦截 #

如果步骤顺序错误（比如把求和步骤提前），TypeScript会直接抛出类型错误，提前拦截运行时风险：

// 错误示例：sumNumbers返回number，而filterValidNumbers需要输入number[]，类型不匹配
// const invalidResult = createPipeline(['1', '2', 'three'], parseNumbers, sumNumbers, filterValidNumbers);
// 报错：Argument of type '(data: number[]) => number' is not assignable to parameter of type 'Transformation<number[], number[]>'.

补充：satisfies运算符的辅助作用 #

如果需要显式确认单个转换函数的类型，可以使用satisfies运算符避免类型断言，同时保留函数的原始类型：

const parseNumbers = (data: string[]) => data.map(Number) satisfies Transformation<string[], number[]>;

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