From 96b9eac4f7c9f388865fffea1a4aa2a05a5ddb0d Mon Sep 17 00:00:00 2001
From: "Kevin C. Coram" <kcoram@gmail.com>
Date: Mon, 12 Oct 2020 12:22:24 -0400
Subject: [PATCH] Implement OnGatewayInit and OnGatewayDisconnect

---
 apps/api/src/app/app.gateway.ts | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/apps/api/src/app/app.gateway.ts b/apps/api/src/app/app.gateway.ts
index cb2dd3d..88c0fd0 100644
--- a/apps/api/src/app/app.gateway.ts
+++ b/apps/api/src/app/app.gateway.ts
@@ -6,15 +6,18 @@ import {
 import { Logger } from '@nestjs/common';
 import {
   OnGatewayConnection,
+  OnGatewayDisconnect,
+  OnGatewayInit,
   SubscribeMessage,
   WebSocketGateway,
   WebSocketServer,
 } from '@nestjs/websockets';
 import { AppService } from './app.service';
-import { Server,Socket } from 'socket.io';
+import { Server, Socket } from 'socket.io';
 
 @WebSocketGateway(4001)
-export class AppGateway implements OnGatewayConnection {
+export class AppGateway
+  implements OnGatewayInit, OnGatewayConnection, OnGatewayDisconnect {
   @WebSocketServer()
   wss: Server;
 
@@ -22,10 +25,19 @@ export class AppGateway implements OnGatewayConnection {
 
   constructor(private appService: AppService) {}
 
+  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());
   }
 
+  handleDisconnect(client: Socket) {
+    this.logger.log(`Client disconnected: ${client.id}`);
+  }
+
   @SubscribeMessage(MESSAGE_TO_SERVER)
   handleMessage(client: Socket, payload: Message): void {
     this.logger.log(payload);