Merge e3b9122 into 8d62307

packages/websockets/interfaces/nest-gateway.interface.ts

@@ -1,5 +1,5 @@
 export interface NestGateway {
   afterInit?: (server: any) => void;
-  handleConnection?: (client: any) => void;
+  handleConnection?: (client: any, ...rest: any[]) => void;
   handleDisconnect?: (client: any) => void;
 }

packages/websockets/interfaces/on-gateway-connection.interface.ts

@@ -1,3 +1,3 @@
 export interface OnGatewayConnection<T = any> {
-  handleConnection(client: T);
+  handleConnection(client: T, ...rest: any[]);
 }

packages/websockets/test/web-sockets-controller.spec.ts

@@ -222,7 +222,7 @@ describe('WebSocketsController', () => {
       ).to.be.a('function');
     });
     it('should call "next" method of connection object with expected argument', () => {
-      expect(nextSpy.calledWith(client)).to.be.true;
+      expect(nextSpy.calledWith([client])).to.be.true;
     });
     it('should call "subscribeMessages" with expected arguments', () => {
       expect(subscribeMessages.calledWith(handlers, client, gateway)).to.be

packages/websockets/web-sockets-controller.ts

@@ -103,8 +103,11 @@ export class WebSocketsController {
     connection: Subject<any>,
   ) {
     const adapter = this.config.getIoAdapter();
-    return client => {
-      connection.next(client);
+    return (client, ...rest) => {
+      // here different websocket implements may have different args
+      // `ws` will have two args applied to this callback
+      connection.next([client, ...rest]);
+
       context.subscribeMessages(messageHandlers, client, instance);
 
       const disconnectHook = adapter.bindClientDisconnect;
@@ -121,9 +124,9 @@ export class WebSocketsController {
 
   public subscribeConnectionEvent(instance: NestGateway, event: Subject<any>) {
     if (instance.handleConnection) {
-      event
-        .pipe(distinctUntilChanged())
-        .subscribe(instance.handleConnection.bind(instance));
+      event.pipe(distinctUntilChanged()).subscribe((args: any[]) => {
+        instance.handleConnection.bind(instance)(...args);
+      });
     }
   }