nestjs websocket 用什么模块

NestJS 是一个用于构建高效、可靠和可扩展的服务器端应用程序的框架。它使用 TypeScript 编写（但也兼容纯 JavaScript），并采用了与 Angular 相似的模块化结构。

在 NestJS 中实现 WebSocket 通信，通常会使用 `@nestjs/platform-express` 包提供的 WebSocket 功能，该平台基于 Express。不过，NestJS 也支持原生 WebSocket 模块 `ws` 或 `@nestjs/websocket`。此外，NestJS 提供了一个专门的模块叫做 `@nestjs/websockets`，它基于 `ws` 包构建，并且提供了更高级别的抽象，使得开发者可以更容易地以声明式的方式处理 WebSocket 消息。

使用 `@nestjs/websockets` 模块，你可以定义 WebSocket 服务器，创建消息处理器，并且将它们与客户端连接进行交互。NestJS 还允许你在依赖注入系统中使用相同的模式，就像你在处理 HTTP 请求时所做的那样。

以下是一个简单的 NestJS WebSocket 实现的例子：

1. 首先，安装必要的依赖：

npm install @nestjs/websockets @nestjs/platform-express

2. 创建 WebSocket 模块：

import { Module } from '@nestjs/common';
import { WebsocketsModule } from '@nestjs/websockets';

@Module({
  imports: [WebsocketsModule],
})
export class WebsocketsModule {}

3. 定义 WebSocket 服务器和消息处理器：

import { Injectable, OnModuleInit, WebSocketServer } from '@nestjs/common';
import { Server } from 'ws';
import { WsConsumer, WsResponse } from '@nestjs/websockets';

@Injectable()
export class WsServer implements OnModuleInit {
  @WebSocketServer() server: Server;

  onModuleInit() {
    this.server.on('connection', (client) => {
      console.log('Client connected', client);
    });
  }

  @Injectable()
  class MessageConsumer implements WsConsumer {
    handleOpen(client: WebSocket) {
      // Handle client connection
    }

    handle закрытие(client: WebSocket) {
      // Handle client disconnection
    }

    handleMessage(client: WebSocket, message: any): WsResponse {
      // Handle incoming messages
    }
  }
}

通过这种方式，你可以定义如何处理 WebSocket 连接的打开、关闭以及接收到的消息等事件。