TypeScript:多维数组参数的正确方法

Question

我正在使用 TS 处理一些 JS 算法。所以我做了这个功能:

function steamrollArray(arr: any[]): any[] {
  return arr.reduce(
    (accum: any, val: any) =>
      Array.isArray(val) ? accum.concat(steamrollArray(val)) : accum.concat(val)
  , []);
}

但参数需要灵活地接受多维数组，如下所示:

steamrollArray([[["a"]], [["b"]]]);
steamrollArray([1, [2], [3, [[4]]]]);
steamrollArray([1, [], [3, [[4]]]]);
steamrollArray([1, {}, [3, [[4]]]]);

定义函数参数的正确方法是什么？

当然我可以使用类型，就像这里:typescript multidimensional array with different types但不适用于所有情况。

Answer 1

您需要定义一个类型，它可能是数组，也可能不是。像这样的东西:

type MaybeArray<T> = T | T[];

然后您可以将函数更新为:

function steamrollArray<T>(arr: MaybeArray<T>[]): T[] {
    return arr.reduce(
        (accum: any, val: any) =>
             Array.isArray(val) ? accum.concat(steamrollArray(val)) : accum.concat(val)
    , []);
}

有了它，Typescript 将能够正确解析类型并理解您的意图。