feat(TCP & WebSocket severs): Add onConnected and onDisconnected methods

src/tcp/TcpServer.ts

@@ -30,11 +30,12 @@ export class TcpServer extends StreamServer {
     })
   }
 
-  protected onConnection(client: Socket): void {
+  protected onConnected(client: Socket): void {
     this.clients.push(client)
-    client.on('close', () => {
-      this.clients.splice(this.clients.indexOf(client), 1)
-    })
+  }
+
+  protected onDisconnected(client: Socket): void {
+    this.clients.splice(this.clients.indexOf(client), 1)
   }
 
   public async start(executor?: Executor): Promise<void> {
@@ -48,7 +49,10 @@ export class TcpServer extends StreamServer {
           )
         })
       }))
-      server.on('connection', client => this.onConnection(client))
+      server.on('connection', client => {
+        this.onConnected(client)
+        client.on('close', () => this.onDisconnected(client))
+      })
 
       const { host, port } = this.address
       return new Promise(resolve => server.listen(port, host, () => resolve()))

src/ws/WebSocketServer.ts

@@ -38,8 +38,9 @@ export class WebSocketServer extends HttpServer {
         socket.on('message', async (message: string) => {
           socket.send(await this.receive(message))
         })
-        // Register connection
-        this.onConnection(connection)
+        // Register connection and disconnection handler
+        this.onConnected(connection)
+        socket.on('close', () => this.onDisconnected(connection))
       },
       options: {
         verifyClient: authorize