Compare commits

..

2 Commits

1 changed files with 16 additions and 4 deletions

View File

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