从函数参数推断联合类型

Question

在 Typescript 中，当我检查变量时，然后分支会根据条件推断变量类型：

type MyType = 'val1' | 'val2' | 'val3'

const variable = 'val1' as MyType

if (variable === 'val2' || variable === 'val3') {
    // typeof variable == 'val2' | 'val3'
}

上面的代码很啰嗦，尤其是当我检查两个以上的值并且我想用它替换它时

if (hasOneOf(variable, 'val2', 'val3')) {
    // typeof variable == 'val2' | 'val3'
}

我部分满意的解决方案是：

const hasOneOf = <T, U extends T>(value: T | undefined, ...values: U[]): value is U => value !== undefined && values.indexOf(value as U) >= 0

但是只有当我明确定义类型时它才有效

if (hasOneOf(variable, 'val2', 'val3')) {
    // typeof variable == MyType
}

// vs

if (hasOneOf<MyType, 'val2' | 'val3'>(variable, 'val2', 'val3')) {
    // typeof variable == 'val2' | 'val3'
}

是否可以创建函数 hasOneOf 来推断泛型类型并返回由参数定义的联合 ('val2', 'val3' => 'val2' | 'val3') ？

Answer 1

如果您正在寻找一个接受可变数量的联合值的解决方案，以下是您问题中的一个变体。您可以将readonly 数组作为第二个参数传递，编译器将根据其成员的类型创建一个联合。 （我认为添加数组括号和as const 断言仍然比提供类型参数更简洁）：

TS Playground link

function isInUnion <T, U extends T>(value: T, values: readonly U[]): value is U {
  return values.indexOf(value as U) >= 0;
}

let str = 'val1'; // string

if (isInUnion(str, ['val2', 'val3'] as const)) {
  str // "val2" | "val3"
}

let num = 1; // number

if (isInUnion(num, [2, 3] as const)) {
  num // 2 | 3
}

---

原答案：

您可以为此使用type predicate：

TS Playground link

function isInUnion <S1 extends string, S2 extends string>(
  value: string,
  str1: S1,
  str2: S2,
): value is S1 | S2 {
  return value === str1 || value === str2;
}

let str = 'val1'; // string

if (isInUnion(str, 'val2', 'val3')) {
  // in this block, str is "val2" | "val3"
}