方法的泛型类型取决于 TypeScript 中类的泛型类型

Question

我有一个用例可以用伪代码表示如下：

class Gen<T> {
  public doStuff<U>(input: U 
         /* If T is an instance of number, 
          then input type U should be an instance of custom type ABC, or
          If T is an instance of string,
          then input type U should an instance of custom type XYZ, else
          compile error */) {

     // do stuff with input
  }
}

这可以用 TypeScript 表达吗？

Answer 1

当然，通过推理，这很容易实现。 TypeScript 允许您基于通用输入类型检查“返回”不同的类型。你应该像这样使用InferInputType<T> 类型：

type InferInputType<T> =
    T extends number ? ABC :
    T extends string ? XYZ :
    never;

然后您可以将您的 Gen 重写为：

class Gen<T> {
    public doStuff(input: InferInputType<T>) {}
}

然后你可以像这样使用你的类：

const genNumber = new Gen<number>();
genNumber.doStuff({ value: 10 });
genNumber.doStuff({ value: 'abc' }); // Error

您可以在操场上看到一个工作示例：Playground Link