打字稿：Es6 Map TS 类型关系

Question

是否可以在 Map 的 Key/Value 类型之间创建某种关系？

我的意思是，如果我使用一个对象，很容易创建类似的东西：

type TObject = { key: string; value: number } | { key: number; value: string }
const obj = { key: 1, value: 'Something' } // Perfect
const obj = { key: 'Something', value: 1 } // Perfect
const obj = { key: 1, value: 1 } // Error

但是，我不知道如何为地图实现此行为

const map = new Map<string | number, number | string>();  
map.set(1, 'a'); // Perfect
map.set('a', 1); // Perfect
map.set(1, 1); // I want to display an error, with the type above : this will works since <number, number> is a possible type
map.set('a', 'a'); // I want to display an error, currently this will works since <string, string> is a possible type

Answer 1

定义一个覆盖Map 的自定义类型，以将调用签名指向您想要的。

interface CustomMap<T, U> extends Map<T | U, T | U> {
  get(key: T): U | undefined;
  get(key: U): T | undefined;
  set(key: T, value: U): this;
  set(key: U, value: T): this;
}

const map: CustomMap<string, number> = new Map();
map.set(1, 'a'); // OK
map.set('a', 1); // OK
map.set(1, 1); // error as expected
map.set('a', 'a'); // error as expected
map.size; // OK, not overridden, still a number
map.delete(1); // OK
map.delete('a'); // OK