强制使用泛型类型的值的类型安全

Question

如何使用泛型将值的类型强制为特定类型？

// An example array
const testArr = [
  {
    id: 3,
    name: 'Spaghetto', // NOTE: Type 'string' here
    shouldNotWork: 3.14, // NOTE: Type 'number' here
  },
  {
    id: 5,
    name: 'Bread',
    shouldNotWork: 3.14,
  },
];

这就是我试图成为我的映射功能的原因，但是我必须附加as V2才能使TS不抱怨：/

type Mapping<T, U> = (val: T, i: number, arr: T[]) => U;

interface Option<T> {
  value: T;
  label: string; // <- NOTE: Type string is required
}

const typeToOption = <
  T,
  K1 extends keyof T,
  K2 extends keyof T,
  V2 extends T[K2] & string // <- NOTE: 'string' union here to match
>(
  valueK: K1,
  labelK: K2,
): Mapping<T, Option<T[K1]>> => (item: T): Option<T[K1]> => ({
  value: item[valueK],
  label: item[labelK] as V2,
});

我希望TS允许我这样做

const result = testArr.map(typeToOption('id', 'name'));

...但不是这个

const result = testArr.map(typeToOption('id', 'shouldNotWork'));

如何让TS投诉后者？

Answer 1

我认为您想像这样键入typeToOption() ：

const typeToOption = <
  T extends Record<K2, string>,
  K1 extends keyof T,
  K2 extends keyof T,
>(
  valueK: K1,
  labelK: K2,
): Mapping<T, Option<T[K1]>> => (item: T): Option<T[K1]> => ({
  value: item[valueK],
  label: item[labelK],
});

使用TypeScript，您可以在通用参数中执行一些“圆形”类型约束。 因此，在这种情况下，我们只需要T extends Record<K2, string> ，而不是完全指定V2 （因为它没有做任何工作），这意味着类型T必须在键处具有string值的属性。 K2 这足以给您您要的行为：

const result = testArr.map(typeToOption('id', 'name')); // okay

const result = testArr.map(typeToOption('id', 'shouldNotWork')); // error
// [ts] Argument of type '"id"' is not assignable to 
// parameter of type '"shouldNotWork"'.

尽管错误有点令人困惑，但至少是错误。 （如果要紧：类型推断失败会导致有趣的事情发生T退回到Record<'shouldNotWork', string> ，等效于{shouldNotWork: string} ，并且keyof T的键仅是'shouldNotWork' ，因此K1 extends ' shouldNotWork'`。噢。）

希望能有所帮助； 祝好运！