如何通过联合类型字符串字面量实现TypeScript自动类型推断？

核心思路 #

要让TypeScript自动推断Service和DTO的对应关系，关键是建立服务名称、DTO、Service类三者的强类型关联，避免松散联合类型导致的推断失效。通过类型映射表+泛型约束就能实现类型安全的统一调用，不用逐个判断属性。

具体实现代码 #

1. 定义基础DTO和Service类 #

// 数据传输对象
class UserDto {
  name: string;
  age: number;
}

class CarDto {
  brand: string;
  model: string;
}

// 对应服务类
class UserService {
  create(dto: UserDto): void {
    console.log('创建用户:', dto);
  }
}

class CarService {
  create(dto: CarDto): void {
    console.log('创建车辆:', dto);
  }
}

2. 建立类型映射表 #

定义ServiceMap类型明确每个服务名称对应的DTO和Service类，这是实现自动推断的核心：

// 服务与DTO的强关联映射
type ServiceMap = {
  userService: {
    dto: UserDto;
    service: UserService;
  };
  carService: {
    dto: CarDto;
    service: CarService;
  };
};

// 提取所有合法的服务名称联合类型
type ValidServiceName = keyof ServiceMap;

// 定义方法入参类型，根据服务名称自动匹配DTO
type ServiceCallParams<T extends ValidServiceName> = {
  service: T;
  dto: ServiceMap[T]['dto'];
};

3. 实现MainService的统一管理方法 #

用泛型方法约束入参，让TypeScript自动推断对应的Service和DTO类型：

class MainService {
  // 初始化所有服务实例
  userService = new UserService();
  carService = new CarService();

  // 泛型管理方法，自动校验类型
  serviceManager<T extends ValidServiceName>(params: ServiceCallParams<T>): void {
    // 类型断言明确当前服务的具体类型
    const targetService = this[params.service] as ServiceMap[T]['service'];
    // 此时TypeScript已能推断create方法的参数类型完全匹配
    targetService.create(params.dto);
  }
}

4. 调用示例（类型安全） #

const mainService = new MainService();

// 正确调用：自动推断dto必须符合UserDto结构
mainService.serviceManager({
  service: 'userService',
  dto: { name: '张三', age: 25 }
});

// 正确调用：自动推断dto必须符合CarDto结构
mainService.serviceManager({
  service: 'carService',
  dto: { brand: '特斯拉', model: 'Model 3' }
});

// 类型错误：dto不符合UserDto要求，TypeScript会直接报错
// mainService.serviceManager({
//   service: 'userService',
//   dto: { brand: '特斯拉' }
// });

扩展说明 #

如果需要新增服务，只需要：

1. 新增对应的DTO和Service类
2. 在ServiceMap中添加新的映射条目
3. 泛型约束会自动更新，无需修改serviceManager方法逻辑

替代方案（基于原DataType联合类型） #

如果想保留原有的DataType联合类型，也可以通过条件类型实现推断：

// 原DataType联合类型
type DataType =
  | { type: 'user', service: 'userService', dto: UserDto }
  | { type: 'car', service: 'carService', dto: CarDto };

class MainService {
  userService = new UserService();
  carService = new CarService();

  serviceManager<T extends DataType>(params: T): void {
    // 用条件类型断言服务类型
    const targetService = this[params.service] as 
      T extends { service: 'userService' } ? UserService :
      T extends { service: 'carService' } ? CarService : never;
    targetService.create(params.dto);
  }
}

这种方式适合已有联合类型的场景，但扩展性不如类型映射表（新增服务时需要修改条件判断）。

内容的提问来源于stack exchange，提问作者Emi