如何正确为TypeScript泛型添加约束条件

问题概述 #

我创建了CascadeStrategies类用于管理一组策略函数，期望所有策略函数使用相同参数集并返回string类型，但在调整泛型约束后遇到了类型兼容问题：

原始代码 #

export class CascadeStrategies<
  T extends Record<any, (...args: any[]) => unknown>
> {
  private strategies: T = {} as T;

  constructor(strategyMap: T) {
    this.registerStrategies(strategyMap);
  }

  private registerStrategies(strategyMap: T) {
    this.strategies = strategyMap;
  }

  use(
    strategies: (keyof T)[],
    ...args: Parameters<T[keyof T]>
  ): ReturnType<T[keyof T]> {
    return this.strategies[strategies[0]](...args);
  }
}

预期用法 #

const myMap = {
  test: (arg1: number, arg2: string) => arg1,
  otherTest: (arg1: number, arg2: string) => arg2,
  thirdTest: (arg1: number, arg2: string) => null
}
const cascadeStrats = new CascadeStrategies(myMap);
const shouldBeNumber = cascadeStrats.use(["test"], 0, "");
const shouldBeString = cascadeStrats.use(["otherTest"], 0, "");
const shouldBeNull = cascadeStrats.use(["thirdTest"], 0, "");

当我将泛型约束改为T extends Record<any, (...args: unknown[]) => string>以强制所有策略返回string时，this.strategies[strategies[0]](...args)的类型被推断为unknown，与ReturnType<T[keyof T]>不兼容。虽然将内部strategies改为Record<any, any>能临时解决问题，但我希望满足三个核心需求：

• 用户定义strategyMap时能获得严格的类型检查（所有函数参数一致、返回string）
• 内部strategies避免使用any类型
• 调用use方法时参数和返回值类型能被正确推断

解决方案 #

通过拆分泛型参数，将策略函数的参数类型和返回值类型单独提取出来，让TypeScript能明确追踪类型关系，避免推断歧义。

方案1：统一参数与返回值类型的简化版本 #

此版本直接约束所有策略使用相同参数和返回值，类型推断更清晰：

export class CascadeStrategies<Args extends unknown[], Return = string> {
  private strategies: Record<string, (...args: Args) => Return>;

  constructor(strategyMap: Record<string, (...args: Args) => Return>) {
    this.strategies = strategyMap;
  }

  use(strategies: string[], ...args: Args): Return {
    return this.strategies[strategies[0]](...args);
  }
}

方案2：保留键名类型校验的增强版本 #

如果需要限制use方法的strategies参数只能传入已定义的策略键名，可以使用双泛型约束：

export class CascadeStrategies<
  T extends Record<string, (...args: Args) => Return>,
  Args extends unknown[],
  Return = string
> {
  private strategies: T;

  constructor(strategyMap: T) {
    this.strategies = strategyMap;
  }

  use(strategies: (keyof T)[], ...args: Args): ReturnType<T[keyof T]> {
    // 类型断言：通过泛型约束已保证所有策略返回值类型一致，此处断言消除TypeScript的推断歧义
    return this.strategies[strategies[0]](...args) as ReturnType<T[keyof T]>;
  }
}

验证与说明 #

类型检查效果 #

对于符合约束的策略对象，TypeScript会自动推断参数和返回值类型：

const myMap = {
  test: (arg1: number, arg2: string) => arg1.toString(),
  otherTest: (arg1: number, arg2: string) => arg2,
  thirdTest: (arg1: number, arg2: string) => "null"
}
const cascadeStrats = new CascadeStrategies(myMap);

// 返回值均被推断为string
const str1 = cascadeStrats.use(["test"], 0, "");
const str2 = cascadeStrats.use(["otherTest"], 0, "");
const str3 = cascadeStrats.use(["thirdTest"], 0, "");

如果传入参数不一致或返回值不符合要求的策略对象，TypeScript会直接报错：

// 错误：参数数量不匹配
const badMap = {
  test: (arg1: number) => "test",
  otherTest: (arg1: number, arg2: string) => "other"
}
const badStrats = new CascadeStrategies(badMap);

满足核心需求 #

1. 用户侧类型推断：通过泛型约束强制所有策略函数参数、返回值一致，不符合要求的输入会立即报错；
2. 内部类型安全：strategies使用具体泛型类型，未使用any；
3. use方法类型推断：参数和返回值类型会自动从策略对象中推导，调用时获得完整的类型提示。

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