this return for server/client instance methods

Author: nucleardreamer

lib/websocket/client.js

@@ -68,6 +68,7 @@ class WebSocketClient extends EventEmitter {
     if (packed) {
       this.client.send(packed)
     }
+    return this
   }
 
   /**
@@ -85,6 +86,7 @@ class WebSocketClient extends EventEmitter {
     _.forEach(ns, function (n) {
       _this.pub('sub', n)
     })
+    return this
   }
 
   /**
@@ -103,6 +105,7 @@ class WebSocketClient extends EventEmitter {
     _.forEach(ns, function (n) {
       _this.pub('unsub', n)
     })
+    return this
   }
 
   /**
@@ -115,6 +118,7 @@ class WebSocketClient extends EventEmitter {
         keepClosed: true
       })
     }
+    return this
   }
 }
 

lib/websocket/server.js

@@ -161,6 +161,7 @@ class WebSocketServer extends EventEmitter {
         }
       })
     }
+    return this
   }
 
   /**
@@ -170,8 +171,9 @@ class WebSocketServer extends EventEmitter {
    */
   close (cb = function () {}) {
     if (this.server) {
-      return this.server.close(cb)
+      this.server.close(cb)
     }
+    return this
   }
 
 }