TypeScript中如何调整Array.reduce的返回类型与入参类型不同？

当然可以修改Array.reduce的预期返回类型，具体方案如下：

核心原理 #

TypeScript的Array.reduce内置了泛型支持，签名为reduce<T>(callback: (accumulator: T, current: ElementType, index: number, array: ElementType[]) => T, initialValue: T): T，只要显式指定泛型T的类型，就能让reduce返回和数组成员类型不同的结果。

你当前代码的问题 #

你没有传入reduce的第二个参数（初始累加器值），因此TS默认将累加器类型推断为和数组成员一致的RelationalReversionQuery，你手动给累加器加了联合类型后，最终返回结果的类型依然是RelationalReversionQuery | RelationalReversionQuery[][]，无法直接赋值给RelationalReversionQuery[][]类型的变量。

解决方法 #

方法1（推荐：类型安全） #

给reduce传入初始累加值，同时显式指定泛型，调整逻辑去掉累加器类型判断的分支，代码如下：

let executionSequence: RelationalReversionQuery[][] = this.reduce<RelationalReversionQuery[][]>(
    (accumulator: RelationalReversionQuery[][], item: RelationalReversionQuery) => {
    if (item.ExecutionSequence !== RelationalReversionQueryExecutionSequence.Indifferent) {
        accumulator.push([item]);
        return accumulator;
    }
    if (accumulator.length > 0) {
        let lastArray = accumulator[accumulator.length - 1];
        if (lastArray.length > 0) {
            if (lastArray[lastArray.length - 1].ExecutionSequence === RelationalReversionQueryExecutionSequence.Indifferent) {
                lastArray.push(item);
            } else {
                accumulator.push([item]);
            }
        } else {
            lastArray.push(item);
        }
    } else {
        accumulator.push([item]);
    }
    return accumulator;
}, []); // 初始值传入空二维数组

方法2（适配现有逻辑） #

如果你的业务逻辑必须保留第一次累加器为第一个数组元素的规则（也就是不传入初始值的写法），可以在reduce调用结束后显式做类型断言，告知TS最终返回结果就是二维数组类型：

let executionSequence: RelationalReversionQuery[][] = this.reduce(
    (accumulator: RelationalReversionQuery|RelationalReversionQuery[][], item: RelationalReversionQuery) => {
    if (accumulator instanceof RelationalReversionQuery) {
        if ((accumulator.ExecutionSequence === RelationalReversionQueryExecutionSequence.Indifferent) && (item.ExecutionSequence === RelationalReversionQueryExecutionSequence.Indifferent)) {
            return [[accumulator, item]]
        }
        return [[accumulator], [item]];
    }
    if (item.ExecutionSequence !== RelationalReversionQueryExecutionSequence.Indifferent) {
        accumulator.push([item]);
        return accumulator;
    }
    if (accumulator.length > 0) {
        let lastArray = accumulator[accumulator.length - 1];
        if (lastArray.length > 0) {
            if (lastArray[lastArray.length - 1].ExecutionSequence === RelationalReversionQueryExecutionSequence.Indifferent) {
                lastArray.push(item);
            } else {
                accumulator.push([item]);
            }
        } else {
            lastArray.push(item);
        }
    } else {
        accumulator.push([item]);
    }
    return accumulator;
}) as RelationalReversionQuery[][];

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