如何在TypeScript的QueryFactory泛型类中强制响应类型匹配预期？

我在TypeScript中定义了一个QueryFactory泛型类，它接收TQuery和TResponse两个泛型类型参数。我希望实现类型安全约束，当传入的TResponse类型与TQuery类型的预期响应类型不匹配时，编译器能抛出错误。

以下是我的代码：

// object examples

class Person {
  name: string;
  age: number;
  gender: string;
}

class Book {
  name: string;
  author: string;
}

// the classes 

interface IBaseRequest {}

class RequestBase<T> implements IRequest<T>, IBaseRequest { }

interface IRequest<TResponse> extends IBaseRequest { }  

class GetPersonByIdQuery extends RequestBase<Person> implements IRequest<Person>, IBaseRequest { }

class QueryFactory<TQuery, TResponse> { }

// the implementation

new QueryFactory<GetPersonByIdQuery, Person>(); // this is allowed as expected.

new QueryFactory<GetPersonByIdQuery, Book>(); // This should not be allowed by the compiler
new QueryFactory<GetPersonByIdQuery, string>(); // This should not be allowed by the compiler

我需要确保传递给QueryFactory的TResponse类型与TQuery类型的预期响应类型一致。例如，GetPersonByIdQuery只能允许Person作为响应类型。请问在TypeScript中是否有办法实现这种类型安全？该如何修改QueryFactory类来达成这一目标？

可以实现这种类型安全约束，核心思路是从TQuery中提取它关联的响应类型，再强制TResponse必须和这个提取出的类型完全匹配。具体实现步骤如下：

1. 定义类型工具提取Query的响应类型 #

先写一个类型工具，用TypeScript的条件类型+类型推断，从TQuery中取出它实现的IRequest接口对应的泛型参数（也就是预期的响应类型）：

type ExtractResponseType<T extends IBaseRequest> = 
  T extends IRequest<infer TResp> ? TResp : never;

2. 修改QueryFactory的泛型约束 #

提供两种实现方式，可根据需求选择：

方式一：自动推导响应类型（无需手动传TResponse） #

这种方式最简洁，用户只需要传入TQuery，响应类型会自动被推导出来，完全避免手动传错的可能：

class QueryFactory<TQuery extends IBaseRequest> {
  // 类内部需要用到响应类型时，直接用ExtractResponseType<TQuery>即可
  responseType!: ExtractResponseType<TQuery>;
}

// 使用示例：自动推导响应类型为Person，无需手动指定
new QueryFactory<GetPersonByIdQuery>(); 

方式二：保留双泛型但添加强约束 #

如果必须保留TResponse参数，可以给它添加约束，同时额外加一个类型断言确保是完全匹配（避免子类型兼容导致的宽松匹配）：

class QueryFactory<TQuery extends IBaseRequest, TResponse extends ExtractResponseType<TQuery>> {
  // 强制TResponse与提取的类型完全一致，避免子类型被允许
  private enforceExactMatch: TResponse extends ExtractResponseType<TQuery> 
    ? ExtractResponseType<TQuery> extends TResponse 
      ? true 
      : false 
    : false = true as any;
}

// 使用示例：
new QueryFactory<GetPersonByIdQuery, Person>(); // 正常通过编译

// 以下两种写法都会触发编译错误，符合预期
new QueryFactory<GetPersonByIdQuery, Book>(); 
new QueryFactory<GetPersonByIdQuery, string>(); 

3. 优化原有接口定义（可选） #

原代码中RequestBase和IRequest的实现存在冗余，可以简化为：

interface IBaseRequest {}

interface IRequest<TResponse> extends IBaseRequest {}  

class RequestBase<TResponse> implements IRequest<TResponse> {}

class GetPersonByIdQuery extends RequestBase<Person> {}

简化后逻辑更清晰，也不影响类型提取的功能。

内容的提问来源于stack exchange，提问作者Mo D Genesis