support close

README.md

@@ -59,7 +59,7 @@ WIP
 
 ### Server events
 
-- ❌ close
+- ✅ close
 - ✅ connection
 - ❌ error
 - ❌ headers
@@ -70,15 +70,15 @@ WIP
 
 - ❌ server.address()
 - ❌ server.clients
-- ❌ server.close(callback)
+- ✅ server.close(callback)
 - ❌ server.handleUpgrade(request, socket, head, callback)
 - ❌ server.shouldHandle(request)
 
 ### Client events
 
 - ❌ close
 - ❌ error
-- ❌ message
+- ✅ message
 - ❌ ping
 - ❌ pong
 - ❌ redirect

src/server.js

@@ -82,4 +82,12 @@ module.exports = class WebSocketServer extends EventEmitter {
             if(callback) callback(this.port);
         });
     }
+
+    close(callback) {
+        this.uwsApp.close();
+        process.nextTick(() => {
+            this.emit("close");
+            if(callback) callback();
+        });
+    }
 }

tests/tests/server/close.js

@@ -0,0 +1,22 @@
+// must support close event
+
+const ws = require("ws");
+
+const wss = new ws.WebSocketServer({ port: 8080 }, cb);
+
+wss.on("close", () => {
+    console.log("Closed server");
+    process.exit(0);
+});
+
+function cb() {
+    const c = new ws.WebSocket("ws://localhost:8080");
+    c.on("open", () => {
+        console.log("Connected to server");
+        c.close();
+    });
+    c.on("close", () => {
+        console.log("Closed connection");
+        process.exit(0);
+    });
+};