apps/api/src/app/app.gateway.ts

@@ -6,18 +6,15 @@ import {
 import { Logger } from '@nestjs/common';
 import {
   OnGatewayConnection,
-  OnGatewayDisconnect,
-  OnGatewayInit,
   SubscribeMessage,
   WebSocketGateway,
   WebSocketServer,
 } from '@nestjs/websockets';
+import { Server } from 'http';
 import { AppService } from './app.service';
-import { Server, Socket } from 'socket.io';
 
 @WebSocketGateway(4001)
-export class AppGateway
+export class AppGateway implements OnGatewayConnection {
-  implements OnGatewayInit, OnGatewayConnection, OnGatewayDisconnect {
   @WebSocketServer()
   wss: Server;
 
@@ -25,21 +22,12 @@ export class AppGateway
 
   constructor(private appService: AppService) {}
 
-  afterInit(server: Server) {
+  handleConnection(client: any, ...args: any[]) {
-    this.logger.log('Initialized...');
-  }
-
-  handleConnection(client: Socket, ...args: any[]) {
-    this.logger.log(`Client connected: ${client.id}`);
     client.emit(MESSAGE_TO_CLIENT, this.appService.getMessages());
   }
 
-  handleDisconnect(client: Socket) {
-    this.logger.log(`Client disconnected: ${client.id}`);
-  }
-
   @SubscribeMessage(MESSAGE_TO_SERVER)
-  handleMessage(client: Socket, payload: Message): void {
+  handleMessage(client: any, payload: Message): void {
     this.logger.log(payload);
     this.appService.addMessage(payload);
     this.wss.emit(MESSAGE_TO_CLIENT, this.appService.getMessages());