zmdms-utils 0.0.80 → 0.0.81

package/dist/es/stompClient.js

@@ -0,0 +1,379 @@
+import './node_modules/zt-stompjs/index.js';
+import entry from './node_modules/zt-sockjs-client/lib/entry.js';
+import { getToken } from './auth.js';
+import { exports as ztStompjsExports } from './_virtual/index.js';
+
+var StompClient = /** @class */ (function () {
+    function StompClient(connectWsConfig) {
+        this.isClient = false; // 是否连接上了
+        this.clearIsClient = null;
+        this.clearIsClientNo = null; // 连接10秒内未建立，重新连接
+        this.clearIsClientYes = null; // 连接已建立重新连接
+        this.subscriptions = {}; // 连接通道的返回值存储对象
+        this.subscriptionsFn = {}; // 连接通道的函数存储对象
+        this.successCallbacks = {}; // 连接成功执行的回调存储对象
+        this.errorCallbacks = {}; // 连接失败执行的回调存储对象
+        this.reconnectionNum = 0; // 重连次数
+        this.reconnectiontime = 5000; // 重连时间
+        this.maxReconnectiontime = 60000; // 最大重连时间 60s
+        this.userClose = false; // 是否用户主动关闭
+        this.isInit = false;
+        this.connectWsConfig = connectWsConfig;
+    }
+    // 连接初始化
+    StompClient.prototype.init = function () {
+        var _this = this;
+        var _a, _b, _c, _d;
+        this.isInit = true;
+        var token = this.connectWsConfig.token || getToken();
+        // 建立连接对象，还未发起连接
+        var socket = new entry("".concat(this.connectWsConfig.baseUrl, "/api/zmdms-csc-ztim/ws?token=").concat(token, "&system=").concat(this.connectWsConfig.systemType, "&clientType=").concat(this.connectWsConfig.clientType));
+        // 获取 STOMP 子协议的客户端对象
+        this.client = ztStompjsExports.over(socket);
+        this.isClient = false;
+        if (!this.subscriptions) {
+            this.subscriptions = {};
+        }
+        if (!this.subscriptionsFn) {
+            this.subscriptionsFn = {};
+        }
+        if (!this.successCallbacks) {
+            this.successCallbacks = {};
+        }
+        if (!this.errorCallbacks) {
+            this.errorCallbacks = {};
+        }
+        // 如果存在 证明已经初始化过，直接删除
+        if (this.clearIsClient) {
+            clearTimeout(this.clearIsClient);
+            this.clearIsClient = null;
+        }
+        if (this.clearIsClientNo) {
+            clearTimeout(this.clearIsClientNo);
+            this.clearIsClientNo = null;
+        }
+        if (this.clearIsClientYes) {
+            clearTimeout(this.clearIsClientYes);
+            this.clearIsClientYes = null;
+        }
+        this.clearIsClient = setTimeout(function () {
+            _this.clearIsClient = null;
+            if (!_this.isClient) {
+                console.log("\u8FDE\u63A5\u5DF2\u53D1\u8D77\uFF0C\u4F46\u662F10\u79D2\u540E\u672A\u6210\u529F\u5EFA\u7ACB\u8FDE\u63A5\uFF0C\u624B\u52A8\u65AD\u5F00\u91CD\u8FDE!system=".concat(_this.connectWsConfig.systemType));
+                try {
+                    _this.client.disconnect(function () {
+                        _this.client = null;
+                    });
+                }
+                catch (err) {
+                    console.log("---连接未建立的关闭出错---", err);
+                }
+                _this.reconnectionNum++;
+                var timeInterval = _this.getReconnectionTime();
+                console.log("---\u7B2C".concat(_this.reconnectionNum, "\u6B21\u91CD\u8FDE\uFF01time=").concat(Date.now(), "system=").concat(_this.connectWsConfig.systemType, "---"));
+                if (_this.clearIsClientNo) {
+                    clearTimeout(_this.clearIsClientNo);
+                    _this.clearIsClientNo = null;
+                }
+                _this.clearIsClientNo = setTimeout(function () {
+                    _this.init();
+                }, timeInterval > _this.maxReconnectiontime
+                    ? _this.maxReconnectiontime
+                    : timeInterval);
+            }
+        }, 10000);
+        // 拦截输出的一大堆垃圾信息
+        this.client.debug = null;
+        // if (this.connectWsConfig.debug) {
+        //   this.client.debug = (str: string) => {
+        //     if (this.connectWsConfig.debug) {
+        //       console.log(str);
+        //     }
+        //   };
+        // }
+        // 心跳检测 stompjs 底层使用window.setInterval() 定期发送心跳检测
+        this.client.heartbeat.outgoing =
+            (_b = (_a = this.connectWsConfig.heartbeat) === null || _a === void 0 ? void 0 : _a.outgoing) !== null && _b !== void 0 ? _b : 2000; // 客户端将每2000ms发送一次心跳
+        this.client.heartbeat.incoming =
+            (_d = (_c = this.connectWsConfig.heartbeat) === null || _c === void 0 ? void 0 : _c.incoming) !== null && _d !== void 0 ? _d : 2000; // 客户端不希望从服务器接收心跳
+        this.client.connect({}, 
+        // 连接成功回调
+        function () {
+            _this.isClient = true;
+            if (_this.clearIsClient) {
+                clearTimeout(_this.clearIsClient);
+                _this.clearIsClient = null;
+            }
+            if (_this.clearIsClientNo) {
+                clearTimeout(_this.clearIsClientNo);
+                _this.clearIsClientNo = null;
+            }
+            if (_this.clearIsClientYes) {
+                clearTimeout(_this.clearIsClientYes);
+                _this.clearIsClientYes = null;
+            }
+            _this.reconnectionNum = 0;
+            console.log("---------\u8FDE\u63A5\u6210\u529F time=".concat(Date.now(), " system=").concat(_this.connectWsConfig.systemType, " time=").concat(Date.now(), "--------"));
+            _this.executeSubscribe();
+        }, 
+        // 连接失败回调
+        function (err) {
+            console.log("---------\u8FDE\u63A5\u5931\u8D25 system=".concat(_this.connectWsConfig.systemType, " time=").concat(Date.now(), "--------"));
+            console.log(err);
+            if (_this.userClose) {
+                _this.userClose = false; // 重新初始化 重新计算
+                if (_this.clearIsClient) {
+                    clearTimeout(_this.clearIsClient);
+                    _this.clearIsClient = null;
+                }
+                if (_this.clearIsClientNo) {
+                    clearTimeout(_this.clearIsClientNo);
+                    _this.clearIsClientNo = null;
+                }
+                if (_this.clearIsClientYes) {
+                    clearTimeout(_this.clearIsClientYes);
+                    _this.clearIsClientYes = null;
+                }
+                // 连接失败后，将所有注册过的订阅、回调执行状态置为未执行状态。
+                _this.setExecuteFail();
+                console.log("用户主动关闭");
+                return;
+            }
+            _this.isClient = true;
+            if (_this.clearIsClient) {
+                clearTimeout(_this.clearIsClient);
+                _this.clearIsClient = null;
+            }
+            if (_this.clearIsClientNo) {
+                clearTimeout(_this.clearIsClientNo);
+                _this.clearIsClientNo = null;
+            }
+            if (_this.clearIsClientYes) {
+                clearTimeout(_this.clearIsClientYes);
+                _this.clearIsClientYes = null;
+            }
+            // 连接失败后，将所有注册过的订阅、回调执行状态置为未执行状态。
+            _this.setExecuteFail();
+            // 重连逻辑, 默认3秒后重连
+            // 重连间隔，每隔5秒递增，超过最大间隔时间后，取最大间隔时间重连
+            try {
+                _this.client.disconnect(function () {
+                    _this.client = null;
+                });
+            }
+            catch (err) {
+                console.log("---\u8FDE\u63A5\u5DF2\u5EFA\u7ACB\u7684\u5173\u95ED\u51FA\u9519 time=".concat(Date.now(), "system=").concat(_this.connectWsConfig.systemType, "---"), err);
+            }
+            _this.reconnectionNum++;
+            var timeInterval = _this.getReconnectionTime();
+            console.log("---\u7B2C".concat(_this.reconnectionNum, "\u6B21\u91CD\u8FDE\uFF01time=").concat(Date.now(), "system=").concat(_this.connectWsConfig.systemType));
+            _this.clearIsClientYes = setTimeout(function () {
+                _this.init();
+            }, timeInterval > _this.maxReconnectiontime
+                ? _this.maxReconnectiontime
+                : timeInterval);
+        });
+    };
+    /**
+     * 连接失败后，将所有注册过的订阅、回调执行状态置为未执行状态。
+     * @param isExecute 连接通道、回调的执行状态
+     */
+    StompClient.prototype.setExecuteFail = function () {
+        var _this = this;
+        // 连接失败后，将当前已注册的 连接通道 全部置为未执行状态
+        var subscriptionsFnKeys = Object.keys(this.subscriptionsFn);
+        if (Array.isArray(subscriptionsFnKeys)) {
+            subscriptionsFnKeys.forEach(function (key) {
+                _this.subscriptionsFn[key].isExecute = false;
+            });
+        }
+        // 连接失败后，统一执行连接失败函数
+        var errorCallbackKeys = Object.keys(this.errorCallbacks);
+        if (Array.isArray(errorCallbackKeys)) {
+            errorCallbackKeys.forEach(function (key) {
+                var _a = _this.errorCallbacks[key], isExecute = _a.isExecute, callback = _a.callback;
+                if (!isExecute) {
+                    callback && callback();
+                }
+                _this.errorCallbacks[key].isExecute = true;
+            });
+        }
+        // 将成功回调 的执行状态 重置为未执行状态
+        var successCallbackKeys = Object.keys(this.successCallbacks);
+        if (Array.isArray(successCallbackKeys)) {
+            successCallbackKeys.forEach(function (key) {
+                _this.successCallbacks[key].isExecute = false;
+            });
+        }
+    };
+    /**
+     * 连接成功后，需要将当前已注册的，未执行的连接通道重新执行连接
+     * 还有成功回调
+     * 失败回调的执行状态重置为未执行状态
+     * @returns 执行订阅相关回调
+     */
+    StompClient.prototype.executeSubscribe = function () {
+        var _this = this;
+        // 连接成功后，将当前已注册的，未执行的 连接通道 连接
+        var subscriptionsFnKeys = Object.keys(this.subscriptionsFn);
+        if (Array.isArray(subscriptionsFnKeys)) {
+            subscriptionsFnKeys.forEach(function (key) {
+                var _a = _this.subscriptionsFn[key], isExecute = _a.isExecute, subscriptionsFn = _a.subscriptionsFn;
+                if (!isExecute) {
+                    subscriptionsFn === null || subscriptionsFn === void 0 ? void 0 : subscriptionsFn();
+                    _this.subscriptionsFn[key].isExecute = true;
+                }
+            });
+        }
+        // 连接成功后，统一执行连接成功函数
+        var successCallbackKeys = Object.keys(this.successCallbacks);
+        if (Array.isArray(successCallbackKeys)) {
+            successCallbackKeys.forEach(function (key) {
+                var _a = _this.successCallbacks[key], isExecute = _a.isExecute, callback = _a.callback;
+                if (!isExecute) {
+                    callback && callback();
+                }
+                _this.successCallbacks[key].isExecute = true;
+            });
+        }
+        // 将失败回调 的执行状态 重置为未执行状态
+        var errorCallbackKeys = Object.keys(this.errorCallbacks);
+        if (Array.isArray(errorCallbackKeys)) {
+            errorCallbackKeys.forEach(function (key) {
+                _this.errorCallbacks[key].isExecute = false;
+            });
+        }
+    };
+    // 重连时间计算
+    StompClient.prototype.getReconnectionTime = function () {
+        var _a;
+        var timeInterval = Math.ceil(this.reconnectionNum / 5) *
+            ((_a = this.connectWsConfig.reconnectionTime) !== null && _a !== void 0 ? _a : this.reconnectiontime);
+        return timeInterval;
+    };
+    // 订阅连接成功回调
+    StompClient.prototype.subscribeSuccessCall = function (subscribeId, callback) {
+        this.successCallbacks[subscribeId] = {
+            isExecute: false,
+            callback: callback,
+        };
+    };
+    // 订阅连接失败回调
+    StompClient.prototype.subscribeErrorCall = function (subscribeId, callback) {
+        this.errorCallbacks[subscribeId] = {
+            isExecute: false,
+            callback: callback,
+        };
+    };
+    // 订阅通道逻辑
+    StompClient.prototype.subscribe = function (subscribeId, pipe, callback, headers) {
+        var _this = this;
+        var _a;
+        // 兼容不传callback 的方案
+        if (typeof callback !== "function") {
+            headers = callback;
+            callback = undefined;
+        }
+        var subscriptionsFn = function () {
+            if (_this.client) {
+                var subscription = _this.client.subscribe(pipe, function (response) {
+                    callback && callback(response);
+                }, headers);
+                _this.subscriptions[subscribeId] = subscription;
+            }
+        };
+        if ((_a = this.client) === null || _a === void 0 ? void 0 : _a.connected) {
+            // 如果是连接状态 正常执行函数
+            subscriptionsFn();
+            // 是否已执行
+            this.subscriptionsFn[subscribeId] = {
+                isExecute: true,
+                subscriptionsFn: subscriptionsFn,
+            };
+        }
+        else {
+            // 缓存连接函数
+            this.subscriptionsFn[subscribeId] = {
+                isExecute: false,
+                subscriptionsFn: subscriptionsFn,
+            };
+        }
+    };
+    StompClient.prototype.getSubscriptions = function (key) {
+        var subscriptionKey = this.subscriptions[key];
+        Reflect.deleteProperty(this.subscriptions, key);
+        Reflect.deleteProperty(this.subscriptionsFn, key);
+        return subscriptionKey;
+    };
+    // 关闭连接逻辑
+    StompClient.prototype.disconnect = function (callback) {
+        var _this = this;
+        this.isInit = false;
+        this.userClose = true;
+        this.subscriptions = null;
+        this.subscriptionsFn = null;
+        this.successCallbacks = null;
+        this.errorCallbacks = null;
+        return new Promise(function (resolve, reject) {
+            try {
+                _this.client.disconnect(function () {
+                    _this.client = null;
+                    callback && callback();
+                    resolve(null);
+                });
+            }
+            catch (err) {
+                callback && callback();
+                resolve(null);
+            }
+        });
+    };
+    // 手动重新连接
+    StompClient.prototype.reconnect = function () {
+        var _this = this;
+        this.isInit = false;
+        this.userClose = true;
+        return new Promise(function (resolve, reject) {
+            try {
+                _this.setExecuteFail();
+                _this.client.disconnect(function () {
+                    _this.client = null;
+                    resolve(null);
+                });
+                setTimeout(function () {
+                    _this.userClose = false;
+                    _this.init();
+                }, 400);
+            }
+            catch (err) {
+                resolve(null);
+            }
+        });
+    };
+    // 发送消息
+    StompClient.prototype.send = function (pipe, headers, message) {
+        if (typeof headers === "string") {
+            message = headers;
+            headers = {};
+        }
+        this.client.send(pipe, headers, message);
+    };
+    return StompClient;
+}());
+StompClient.getInstance = (function () {
+    var instance;
+    return function (connectWsConfig) {
+        if (!instance && connectWsConfig) {
+            instance = new StompClient(connectWsConfig);
+            // ws 初始化
+            instance.init();
+        }
+        // 如果没有初始化，那么初始化一下
+        if (instance && !instance.isInit) {
+            instance.init();
+        }
+        return instance;
+    };
+})();
+
+export { StompClient as default };

package/dist/index.d.ts

@@ -11,7 +11,7 @@ export { validatePassword } from './es/passwordValidate.js';
 export { emailValidate, idCardValidate, isChineseValidate, morePhoneValidate, phoneValidate, strLenValidate } from './es/validate.js';
 export { batchDownloadFiles, createDeleteFileLink, createDownloadLink, createOriginalLink, createThumbnailLink, createUploadFileLink, downloadFile, exportBolb, previewFile } from './es/file.js';
 export { calculateAge, parseTime } from './es/parseTime.js';
-export { BASIC_KEY, EMITTER_CCP_KEY, EMITTER_PCC_KEY, EMITTER_TYPE, TOKEN_KEY } from './es/constants.js';
+export { BASIC_KEY, CLIENT_TYPE, EMITTER_CCP_KEY, EMITTER_PCC_KEY, EMITTER_TYPE, TOKEN_KEY } from './es/constants.js';
 export { convert, html, match, pinyin } from 'pinyin-pro';
 export { domToCanvas } from './es/canvas.js';
 export { imgToPdf } from './es/pdf.js';
@@ -21,6 +21,7 @@ export { createWater, getDefaultContent, mergeWaterConfig } from './es/water.js'
 export { initLocale } from './es/locales/i18n.js';
 export { useTranslation } from './es/locales/use-translation.js';
 export { safeT } from './es/locales/t-in-non-react.js';
+export { default as StompClient } from './es/stompClient.js';
 export { Base64 } from 'js-base64';
 export { default as i18n } from 'i18next';
 export { Trans } from 'react-i18next';

package/dist/index.js

@@ -11,7 +11,7 @@ export { validatePassword } from './es/passwordValidate.js';
 export { emailValidate, idCardValidate, isChineseValidate, morePhoneValidate, phoneValidate, strLenValidate } from './es/validate.js';
 export { batchDownloadFiles, createDeleteFileLink, createDownloadLink, createOriginalLink, createThumbnailLink, createUploadFileLink, downloadFile, exportBolb, previewFile } from './es/file.js';
 export { calculateAge, parseTime } from './es/parseTime.js';
-export { BASIC_KEY, EMITTER_CCP_KEY, EMITTER_PCC_KEY, EMITTER_TYPE, TOKEN_KEY } from './es/constants.js';
+export { BASIC_KEY, CLIENT_TYPE, EMITTER_CCP_KEY, EMITTER_PCC_KEY, EMITTER_TYPE, TOKEN_KEY } from './es/constants.js';
 export { convert, html, match, pinyin } from './es/node_modules/pinyin-pro/dist/index.js';
 export { domToCanvas } from './es/canvas.js';
 export { imgToPdf } from './es/pdf.js';
@@ -21,6 +21,7 @@ export { createWater, getDefaultContent, mergeWaterConfig } from './es/water.js'
 export { initLocale } from './es/src/locales/i18n.js';
 export { useTranslation } from './es/src/locales/use-translation.js';
 export { safeT } from './es/src/locales/t-in-non-react.js';
+export { default as StompClient } from './es/stompClient.js';
 export { Base64 } from './es/node_modules/js-base64/base64.js';
 export { default as i18n } from './es/node_modules/i18next/dist/esm/i18next.js';
 export { Trans } from './es/node_modules/react-i18next/dist/es/Trans.js';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zmdms-utils",
-  "version": "0.0.80",
+  "version": "0.0.81",
   "description": "",
   "main": "dist/index.js",
   "module": "dist/index.js",
@@ -25,7 +25,11 @@
     "decimal.js": "^10.4.3",
     "jwt-decode": "3.1.2",
     "mitt": "^3.0.0",
-    "pinyin-pro": "^3.18.2"
+    "pinyin-pro": "^3.18.2",
+    "sockjs-client": "^1.5.1",
+    "stompjs": "^2.3.3",
+    "zt-sockjs-client": "^0.0.2",
+    "zt-stompjs": "^1.1.2"
   },
   "devDependencies": {
     "@rollup/plugin-commonjs": "^24.0.1",

package/src/constants.ts

@@ -24,6 +24,11 @@ export const EMITTER_TYPE = {
   treeSearchSet: "treeSearchSet", // 父应用传递搜索树的值
 };
 
+export const CLIENT_TYPE = {
+  web: 1,
+  mobile: 2,
+};
+
 // token key 值
 // export const TOKEN_KEY = "Zmdms-Auth";
 export const TOKEN_KEY = (window as any).__TOKEN_KEY__ || "Cicoms-Auth";

package/src/index.ts

@@ -57,6 +57,7 @@ export {
   EMITTER_TYPE,
   TOKEN_KEY,
   BASIC_KEY,
+  CLIENT_TYPE,
 } from "./constants";
 
 /**
@@ -81,3 +82,6 @@ export { createWater, mergeWaterConfig, getDefaultContent } from "./water";
 
 // 多语言相关
 export { initLocale, i18n, useTranslation, Trans, safeT } from "./locales";
+
+// 长连接
+export { default as StompClient } from "./stompClient";