访问或将变量传递给Nestjs中的拦截器

Question

我是Nestjs的新手，一直在研究一些解决方案，但没有找到方法。 我的API返回用户列表，如果存在或从DB中读取用户，则可以从缓存中读取用户，并且基于此，我需要设置标题。 我希望拦截器能够处理这一部分。

我的控制器

  @UseInterceptors(CacheOrServiceHeaderInterceptor)
  @Get('users')
  public async users(
      @Query('id') clinicianId: number,
      @Query('name') specialtyName: string,
  ){
    const setFlags: MetaFlags = {setCacheHeader: false, setServiceHeader: false};

    const data = await this.service.getUsers(id, name, setFlags);
    return data;
}

我的服务类

export interface MetaFlags {
  setServiceHeader: boolean
  setCacheHeader?: boolean,
}
:
:
public async getUsers(....){
// code that handles whether cache is used or a service is used to get users
 and based on it, the metaFlags are set. setCacheHeader = true if read from cache or setServiceHeader = true if read from DB.

我的拦截器看起来像是这样的

@Injectable()
export class CacheOrServiceHeaderInterceptor implements NestInterceptor {
  
    public intercept(context: ExecutionContext, next: CallHandler): Observable<any> {
       
        return next
            .handle()
            .pipe(
                map((data) => {
                    const req = context.switchToHttp().getRequest();
    // this is static but I want a condition here if 
                    if(MetaFlags.setCacheHeader)
                    // set xyz header
                     else set abc header
                    req.res.set('x-api-key', 'pretty secure');
                    return data;
                }
                 
            )
        );}}

方法可以将这些标志传递给拦截器或使拦截器读取这些标志值？

Answer 1

我建议返回应将哪些标题设置为控制器返回服务的一部分。如果您最终使用全球状态，则可以在阅读拦截器并将错误的标头读回端点之前获得多个端点覆盖全局状态的请求。类似于

{
  data: regularDataHere,
  header: 'cache' || 'database'
}

拦截器的MAP，您可以读取data对象，然后删除data属性属性，而不是返回它此标题字段到客户端，和读取标题字段，以便您知道要在响应上设置哪个标头。这样的事情：

@Injectable()
export class CacheOrServiceHeaderInterceptor implements NestInterceptor {
  
    public intercept(context: ExecutionContext, next: CallHandler): Observable<any> {
    return next.handler().pipe(
      map((data: { data: DataType, header: 'cache' | 'database' }) => {
        const res = context.switchToHttp().getResponse<ResponseType>();
        if (data.header === 'database') {
          res.setHeader('x-api-key', 'pretty secure');
        } else {
          res.setHeader(xyz, value);
        }
        return data.data;
      })
    );
  }
}