基于输入的 TypeScript 返回类型

Question

我有以下功能:

subtract(sub: number | CalendarWeek): T {
    if (typeof sub === 'number') {
        return new CalendarWeek(...); // Returns a CalendarWeek if sub is a number
    }

    return 1; // Return number if sub is CalendarWeek
}

问题是我需要定义类型 T，使其依赖于提供的 sub 类型。也就是说，如果 sub 是 CalendarWeek，则 T 是 number，如果 sub 是，则 T 是 CalendarWeek 编号。

Answer 1

处理此问题的传统 TypeScript 方法是使用 overloaded functions .在你的情况下，它会是这样的:

subtract(sub: number): CalendarWeek;
subtract(sub: CalendarWeek): number;
subtract(sub: number | CalendarWeek): number | CalendarWeek {
  if (typeof sub === 'number') {
    return new CalendarWeek(...); // Returns a CalendarWeek if sub is a number
  }
  return 1; // Return number if sub is CalendarWeek
}

你可以想象使用 conditional types而不是在 TypeScript 2.8 及更高版本中用重载来表示它，但它可能比你试图到达这里的更高级。

对于感兴趣的各方，该解决方案将如下所示:

subtract<T extends number | CalendarWeek>(sub: T): T extends number ? CalendarWeek : number;
subtract(sub: number | CalendarWeek): number | CalendarWeek {
  if (typeof sub === 'number') {
    return new CalendarWeek(); // Returns a CalendarWeek if sub is a number
  }
  return 1; // Return number if sub is CalendarWeek
}

希望对您有所帮助。祝你好运!