nestjs-websocket 实现简单的聊天室

最新推荐文章于 2024-07-19 17:13:23 发布

passerma

最新推荐文章于 2024-07-19 17:13:23 发布

阅读量1.9k

点赞数 1

分类专栏： nodejs

本文链接：https://blog.csdn.net/passerma/article/details/119851165

版权

nodejs 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

该博客展示了如何利用NestJS框架创建一个WebSocket服务器，实现用户连接、断开时的消息广播，以及监听和处理客户端发送的'message'和'name'事件。代码中包含了WebSocket的初始化、连接管理和消息传递等功能。

摘要由CSDN通过智能技术生成

https://www.passerma.com/chat/

代码地址 https://github.com/passerma/nestjs-websocket

websocket部分代码如下

import {
  OnGatewayConnection, OnGatewayInit, SubscribeMessage, WebSocketGateway, OnGatewayDisconnect, WebSocketServer
} from '@nestjs/websockets';
import { Server, Socket } from 'socket.io';
import { Logger } from '@nestjs/common';

@WebSocketGateway({
  path: '/socket',
  allowEIO3: true,
  cors: {
    origin: /.*/,
    credentials: true
  }
})
export class AppGateway implements OnGatewayInit, OnGatewayConnection, OnGatewayDisconnect {
  private logger: Logger = new Logger('ChatGateway');
  @WebSocketServer() private ws: Server;  // socket实例
  private connectCounts = 0               // 当前在线人数
  private allNum = 0                      // 全部在线人数
  private users: any = {}                 // 人数信息

  /**
   * 初始化
   */
  afterInit() {
    this.logger.log('websocket init ...');
  }

  /**
   * 链接成功
   */
  handleConnection(client: Socket) {
    this.connectCounts += 1
    this.allNum += 1
    this.users[client.id] = `user-${this.connectCounts}`
    this.ws.emit('enter', { name: this.users[client.id], allNum: this.allNum, connectCounts: this.connectCounts });
    client.emit('enterName', this.users[client.id]);
  }

  /**
   * 断开链接
   */
   handleDisconnect(client: Socket) {
    this.allNum -= 1
    this.ws.emit('leave', { name: this.users[client.id], allNum: this.allNum, connectCounts: this.connectCounts  });
  }

  @SubscribeMessage('message')
  /** 
   * 监听发送消息
  */
  handleMessage(client: Socket, data: any): void {
    this.ws.emit('message', {
      name: this.users[client.id],
      say: data
    });
  }

  @SubscribeMessage('name')
  /** 
   * 监听修改名称
  */
  handleName(client: Socket, data: any): void {
    this.users[client.id] = data
    client.emit('name', this.users[client.id]);
  }
}