feat: add close handler

package.json

@@ -17,6 +17,7 @@
   },
   "dependencies": {
     "@rollup/plugin-commonjs": "^25.0.7",
-    "@rollup/plugin-node-resolve": "^15.2.3"
+    "@rollup/plugin-node-resolve": "^15.2.3",
+    "eventemitter3": "^5.0.1"
   }
 }

packages/feiyun/package.json

@@ -1,7 +1,7 @@
 {
   "name": "feiyun",
   "repository": "https://github.com/hxg2050/feiyun.git",
-  "version": "0.2.17",
+  "version": "0.2.18",
   "private": false,
   "type": "module",
   "main": "./src/index.ts",

packages/handler/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@feiyun/handler",
-  "version": "0.0.27",
+  "version": "0.0.28",
   "license": "MIT",
   "type": "module",
   "main": "./src/index.ts",

packages/server/package.json

@@ -2,7 +2,7 @@
     "name": "@feiyun/server",
     "repository": "https://github.com/hxg2050/feiyun.git",
     "private": false,
-    "version": "0.1.1",
+    "version": "0.1.2",
     "type": "module",
     "main": "./src/index.ts",
     "engines": {

packages/server/src/IServer.ts

@@ -5,6 +5,7 @@ export interface IServer {
     start(): void
 
     handlerCallback?: (client: Socket, data: string) => void
+    closeHandlerCallback?: (client: Socket) => void
 
     /**
      * 绑定uid

packages/server/src/feiyun.ts

@@ -46,6 +46,9 @@ export class Feiyun {
       port: this.config.port
     })
     this.server.start()
+    this.server.closeHandlerCallback = (client) => {
+      client.close();
+    }
     this.server.handlerCallback = (client, data) => {
       const ctx = new Context()
       ctx.socket = client

packages/server/src/server.ts

@@ -49,6 +49,7 @@ export class Server implements IServer {
       this.onMessage(this.clientsFromServerWebSocket.get(ws)!, str)
     });
     this.wss.close((ws) => {
+      this.closeHandlerCallback?.(this.clientsFromServerWebSocket.get(ws)!)
       this.clientsFromServerWebSocket.delete(ws);
     });
     // console.log('ws://127.0.0.1:' + this.config.port);
@@ -65,10 +66,7 @@ export class Server implements IServer {
 
   handlerCallback?: (client: Socket, data: string) => void
 
-  private closeHandler?: (client: Socket) => void
-  onClose(callback: (client: Socket) => void) {
-    this.closeHandler = callback
-  }
+  closeHandlerCallback?: (client: Socket) => void
 
   on(path: string, handler: Handler) {
     this.handlers.set(path, handler)

packages/server/src/socket.ts

@@ -1,4 +1,5 @@
 import { IServer } from './IServer';
+import EventEmitter from 'eventemitter3';
 
 const createId = (() => {
   let id = 0
@@ -7,11 +8,12 @@ const createId = (() => {
   }
 })();
 
-export class Socket {
+export class Socket extends EventEmitter {
+
   id = createId();
 
   constructor(public server: IServer) {
-
+    super();
   }
 
   data: Record<string, any> = {}
@@ -45,6 +47,6 @@ export class Socket {
    * 断开链接
    */
   close() {
-
+    this.emit('close', this)
   }
 }