基于可选参数属性函数的打字稿方法返回类型

Question

尝试创建一个简单的实用程序，它可以：

• 按原样返回给定数组
• 或根据给定的可选参数进行转换。 这是代码：

type MapperFn<T, U> =  (val: T) => U;

interface mapperOpts<T,U> {
  cb?: MapperFn<T,U>
}

interface mapper {
  map<T, U, Z extends mapperOpts<T,U>>(arr: Array<T>, opts: Z): Z extends { cb: MapperFn<T,U> } ? U[]: T[];
}

const obj: mapper = {
  map: (arr, { cb }) => {
    if (!cb) return arr;
    return arr.map(cb);
  }
}

const arr: number[] =[1,2,3];

const result = obj.map(arr, {cb: (element) => element.toString() }); // should be typed as `string[]`

const result2 = obj.map(arr, { cb: (element) => element+1 }); // should be typed as `number[]`

const result3 = obj.map(arr, {}); // should be types as `number[]`

但是，我收到了错误：

Type '<T, U, Z extends mapperOpts<T, U>>(arr: T[], { cb }: Z) => T[] | U[]' is not assignable to type '<T, U, Z extends mapperOpts<T, U>>(arr: T[], opts: Z) => Z extends { cb: MapperFn<T, U>; } ? U[] : T[]'.
  Type 'T[] | U[]' is not assignable to type 'Z extends { cb: MapperFn<T, U>; } ? U[] : T[]'.
    Type 'T[]' is not assignable to type 'Z extends { cb: MapperFn<T, U>; } ? U[] : T[]'.

请注意，result 和 result2 被标记为 unknown[]，这可能意味着来自回调函数的参数类型推断无法正常工作。

我错过了什么？

Typescript Playground

Answer 1

generic 类型参数推断的工作方式的细节似乎没有特别好的文档记录，除了 now obsolete TypeScript Language Specification。

但一般情况下，当编译器看到调用签名为（比如）func<T, U, V>(x: F<T, U>, y: G<T, V>): H<U, V>; 形式为c = func(a, b) 的函数时，它需要尝试推断类型参数T、U , 和 V 来自值 a、b 和 c 的类型。为了推断U，编译器需要检查a 和c 的类型，因为参数x 和返回类型都依赖于U。所以x 和返回类型是T 的潜在推理站点。另一方面，尝试使用b 来推断U 的任何东西是没有希望的，因为y 参数的类型根本不引用U。也就是说，y 不是U 的推理站点。

并非所有推理站点都受到平等对待，有些推理站点比其他推理站点更难推理。返回类型通常是较差的推理站点，因为编译器通常不知道预期的返回类型......如果你写了const c = func(a, b);，你要求编译器推断c的类型，所以func()的返回类型不知道。您只能在 c 已经是 const c: SomeType = func(a, b); 等已知类型的情况下使用返回类型。

涉及类型参数的类型函数越复杂，推理站点的用处就越小。对于f<T>(x: T): void 之类的东西，对f(a) 的调用将很容易将T 推断为a 的类型。但是对于像g<T>(x: T[keyof T]): void 这样的东西，几乎不可能从g(a) 推断出T。在前一种情况下，您从相同类型的值推断T。简单的。在后一种情况下，从该类型属性的联合值推断T。不明显你会如何开始。其他类型的函数往往介于这些极端之间。如果遇到问题，最好的办法是简化推理站点中的类型函数。

最后，有些地方根本不用作推理站点。像h<T, U extends T>(u: U): void 这样的函数签名没有T 的推理站点。推断类型参数时不参考Generic constraints。也许您可能希望编译器从传递给h() 的内容中推断出U，然后从U 中推断出T，但事实并非如此。 T 肯定不会被推断出来，会退回到 unknown 之类的东西。

对于出现这种情况的问题，您可以查看microsoft/TypeScript#38183、microsoft/TypeScript#31529，可能还有许多其他问题（我会搜索“推理站点”）。

---

所以，说了这么多，我对你的map() 方法的建议是这样的：

interface Mapper {
  map<T, Z extends MapperOpts<T, any>>(arr: Array<T>, opts: Z):
    Z extends { cb: MapperFn<T, infer U> } ? U[] : T[];
}

在之前的版本中，U 没有合理的推理站点。相反，我们将仅从 arr 和 opts 推断出 T 和 Z。这很可能会成功。由此，我们可以使用Z 通过显式conditional type inference 提取U。

让我们看看它是如何工作的：

const result = obj.map(arr, { cb: (element) => element.toString() }); // string[]
const result2 = obj.map(arr, { cb: (element) => element + 1 }); // number[]
const result3 = obj.map(arr, {}); // number[]

看起来不错！

Playground link to code