如何在NestJS gRPC拦截器中设置响应元数据

在NestJS的gRPC场景下，你需要通过gRPC原生call实例来设置响应元数据，而非尝试通过getResponse()获取的对象操作。具体实现步骤如下：

1. 导入gRPC官方包的Metadata类，用于构建响应元数据
2. 从RPC上下文中提取gRPC原生的call实例
3. 在响应即将发出的时机，调用call实例的sendMetadata方法发送元数据

修改后的完整代码示例：

import { Injectable, NestInterceptor, ExecutionContext, CallHandler } from '@nestjs/common';
import { Observable } from 'rxjs';
import { tap } from 'rxjs/operators';
import { Metadata } from '@grpc/grpc-js';

@Injectable()
export class SomeInterceptor implements NestInterceptor {
  intercept(context: ExecutionContext, next: CallHandler): Observable<any> {
    const ctx = context.switchToRpc();
    const call = ctx.getContext(); // 获取gRPC原生call实例

    return next.handle().pipe(
      tap(() => {
        // 构建响应元数据
        const responseMeta = new Metadata();
        responseMeta.set('custom-header-key', 'custom-header-value');
        // 发送响应元数据
        call.sendMetadata(responseMeta);
      }),
    );
  }
}

关键说明 #

• ctx.getContext()返回的是gRPC原生的调用实例（比如ServerUnaryCall、ServerStreamingCall等），不同调用类型的实例都支持sendMetadata方法
• sendMetadata用于发送响应的头部元数据，必须在响应体发送前调用，因此用tap操作符在Observable发出值（即准备返回响应体）时执行是合理的
• 确保你的项目已安装@grpc/grpc-js依赖（NestJS的gRPC模块默认依赖该包）

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