TypeScript：如何避免重复映射枚举与消息数据类实现泛型解包

你当前的实现里，MessageTypeMap同时定义了类型别名和常量对象，确实存在重复维护的问题。可以通过从单一映射源自动推导类型的方式彻底避免这种重复，具体方案如下：

1. 定义单一映射源 #

只需要维护一个常量对象，把MsgType枚举值和对应的MessageData子类绑定：

const msgTypeToClass = {
  [MsgType.A]: MessageA,
  [MsgType.B]: MessageB
} as const; // as const 确保TypeScript推导出最精确的类型

2. 自动推导类型映射 #

不需要手动写MessageTypeMap类型，直接从上面的常量推导：

type MessageTypeMap = {
  [K in keyof typeof msgTypeToClass]: InstanceType<typeof msgTypeToClass[K]>
};

这里InstanceType会提取类的实例类型，keyof typeof msgTypeToClass会自动获取MsgType的枚举值作为键。

3. 修改unwrapFilter函数 #

用推导出来的类型和单一映射源实现函数，完全消除重复：

function unwrapFilter<T extends MsgType>(message: MessageContainer, msgTypes: T[]): MessageTypeMap[T] | null {
  for (const msgType of msgTypes) {
    const dataClass = msgTypeToClass[msgType];
    if (message.msgType === msgType && message.data instanceof dataClass) {
      return message.data as MessageTypeMap[T];
    }
  }
  return null;
}

方案优势 #

• 仅需维护msgTypeToClass这一处映射，类型自动同步，彻底避免重复代码
• 类型安全：新增或修改消息类型时，只需更新msgTypeToClass，类型会自动推导，不会出现类型与实际类不匹配的错误
• 代码更简洁，减少手动维护的出错概率

内容的提问来源于stack exchange，提问作者Johannes H.