NestJS中如何拦截Socket服务端主动向客户端发送的消息

NestJS WebSocket 网关的内置拦截器默认仅覆盖事件订阅方法的执行链路：只有@SubscribeMessage装饰的方法直接return的值，会被Nest封装为响应消息走拦截器处理；直接调用注入的socketServer或客户端socket实例的emit方法属于Socket.io原生操作，完全脱离Nest的处理管道，不会触发配置的拦截器，这是框架的默认设计，不是bug。

目前有两种成熟方案可以实现主动出站消息的拦截，可根据业务场景选择：

方案1：封装统一发送方法，手动接入Nest拦截器链路 #

适合需要复用现有Nest拦截器逻辑、仅需对部分主动推送消息做拦截的场景，侵入性低，完全兼容Nest原生拦截器的写法。
核心思路是不要直接调用原生emit，所有主动发消息的逻辑都走自定义封装的发送方法，在方法内通过Nest内置的上下文工具手动挂载拦截器链，处理完数据后再调用原生emit发送。
代码实现参考：

import { Injectable, ModulesRef, ExternalContextCreator } from '@nestjs/core';
import { Server, Socket } from 'socket.io';

@Injectable()
export class SocketDispatchService {
  constructor(
    private modulesRef: ModulesRef,
    private externalContextCreator: ExternalContextCreator,
  ) {}

  /**
   * 带拦截器链的消息发送方法
   * @param target 发送目标：可以是socket实例、server实例、to()/in()返回的广播对象
   * @param event 出站事件名
   * @param payload 消息内容
   * @param interceptors 要应用的拦截器列表，和@UseInterceptors传参一致
   */
  async emit(
    target: Socket | Server | ReturnType<Server['to']>,
    event: string,
    payload: any,
    interceptors: Array<new (...args: any[]) => any> = [],
  ) {
    // 构造模拟处理函数，返回待发送的原始数据
    const baseHandler = async () => payload;
    // 用Nest原生工具绑定拦截器链，指定上下文为WebSocket类型
    const handlerWithInterceptors = this.externalContextCreator.create(
      baseHandler,
      null,
      interceptors,
      null,
      'ws',
    );
    // 执行拦截器链拿到处理后的最终数据
    const finalData = await handlerWithInterceptors();
    // 调用原生emit发送
    return target.emit(event, finalData);
  }
}

使用时直接在网关或其他服务里注入这个封装好的服务，替换原生emit调用即可：

// 原来的写法：this.socketServer.to(socket.id).emit("OutgoingEvent", data)
// 替换为：
await this.socketDispatchService.emit(
  this.socketServer.to(socket.id),
  "OutgoingEvent",
  data,
  [SomeInterceptor] // 这里直接传你写好的拦截器类即可，和@UseInterceptors用法完全一致
);

这种方式下你之前写的Nest拦截器不需要做任何修改，拦截逻辑和控制器、默认事件响应的拦截逻辑完全一致，支持依赖注入、全局拦截器等所有Nest拦截器特性。

方案2：Socket.io原生层全局拦截所有出站消息 #

适合需要无差别拦截所有出站消息（包括Nest自动发送的事件响应、所有地方手动调用emit发送的消息）的场景，不需要修改现有emit调用代码。
核心思路是在网关初始化阶段，重写Socket.io实例的emit相关方法，在原生层拦截所有出站数据包做处理：

import { WebSocketGateway, WebSocketServer, OnGatewayInit } from '@nestjs/websockets';
import { Server, Socket } from 'socket.io';
import { OutgoingDataTransformService } from './outgoing-data-transform.service';

@WebSocketGateway()
export class AppGateway implements OnGatewayInit {
  @WebSocketServer()
  private server: Server;

  constructor(
    // 如果需要用Nest容器里的服务处理数据，直接注入即可
    private transformService: OutgoingDataTransformService,
  ) {}

  afterInit(server: Server) {
    // 拦截每个客户端连接的事件
    server.use((socket: Socket, next) => {
      // 1. 重写单socket的emit方法，拦截发给单个客户端的消息
      const originalSocketEmit = socket.emit.bind(socket);
      socket.emit = (event: string, data: any, ...restArgs) => {
        const processed = this.transformService.transform(data);
        return originalSocketEmit(event, processed, ...restArgs);
      };

      next();
    });

    // 2. 重写server的emit方法，拦截全局广播消息
    const originalServerEmit = server.emit.bind(server);
    server.emit = (event: string, data: any, ...restArgs) => {
      const processed = this.transformService.transform(data);
      return originalServerEmit(event, processed, ...restArgs);
    };

    // 3. 重写to/in方法，拦截指定房间/分组的广播消息
    const wrapBroadcastOperator = (broadcastOperator: ReturnType<Server['to']>) => {
      const originalBroadcastEmit = broadcastOperator.emit.bind(broadcastOperator);
      broadcastOperator.emit = (event: string, data: any, ...restArgs) => {
        const processed = this.transformService.transform(data);
        return originalBroadcastEmit(event, processed, ...restArgs);
      };
      return broadcastOperator;
    };

    const originalTo = server.to.bind(server);
    server.to = (room: string) => wrapBroadcastOperator(originalTo(room));
    // in是to的别名，同步重写
    const originalIn = server.in.bind(server);
    server.in = (room: string) => wrapBroadcastOperator(originalIn(room));
  }
}

这个方案的缺点是脱离了Nest原生拦截器体系，无法直接复用已经写好的NestInterceptor实现，需要把处理逻辑抽到单独的服务里调用。

选型建议 #

• 如果只是处理事件订阅方法的响应，直接在方法里return数据即可，Nest默认会走拦截器链，不需要额外改造
• 如果需要对部分主动推送的消息做拦截、且要复用现有Nest拦截器，选方案1
• 如果需要全局拦截所有出站消息、不想改动现有发送逻辑，选方案2

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