Merge pull request #29 from beclab/feat/update-websocket-status

feat

Author: qq815776412

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bytetrade/core",
-  "version": "0.3.84",
+  "version": "0.3.85",
   "description": "didvault core module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",

src/websocket/WebSocketHeart.ts

@@ -10,13 +10,15 @@ export class WebSocketHeart implements IWebSocketHeart {
 	heartGapTime: number;
 	failNum = 0;
 	heartFailNum: number;
+	heartRes?: (ev: any) => boolean;
 
 	constructor(websocketbean: IWebSocketBean) {
 		this.websocketbean = websocketbean;
 		this.heartSend = this.websocketbean.param.heartSend ?? 'heartSend';
 		this.heartGet = this.websocketbean.param.heartGet ?? 'heartGet';
 		this.heartGapTime = this.websocketbean.param.heartGapTime ?? 30000;
 		this.heartFailNum = this.websocketbean.param.heartFailNum ?? 10;
+		this.heartRes = this.websocketbean.param.heartRes;
 	}
 
 	timer: number = null as any;
@@ -52,6 +54,12 @@ export class WebSocketHeart implements IWebSocketHeart {
 	 * @param ev - The received message.
 	 */
 	onmessage = (ev: any) => {
+		if (this.heartRes) {
+			if (this.heartRes(ev)) {
+				this.failNum = 0;
+			}
+			return;
+		}
 		const messagePrefix = this.websocketbean.param.messagePrefix ?? '';
 		const messageSuffix = this.websocketbean.param.messageSuffix ?? '';
 		const heartGetMessage = messagePrefix + this.heartGet + messageSuffix;

src/websocket/websocket.ts

@@ -220,6 +220,8 @@ export interface IWebSocketHeart {
 	 * @returns
 	 */
 	onmessage: (ev: any) => any;
+
+	heartRes: (ev: any) => boolean;
 }
 
 /**