优化diamante

public/components/frame/nav/nav.js

@@ -208,68 +208,106 @@
 
     vm.getUserInfo();
 
-    //建立websocket 消息连接
-    let _userId = vc.getData('/nav/getUserInfo').userId;
-
-    let _protocol = window.location.protocol;
-    let url = '';
-    if (_protocol.startsWith('https')) {
-        url =
-            "wss://" + window.location.host + "/ws/message/" +
-            _userId;
-    } else {
-        url =
-            "ws://" + window.location.host + "/ws/message/" +
-            _userId;
-    }
-
-
-    if ("WebSocket" in window) {
-        websocket = new WebSocket(url);
-    } else if ("MozWebSocket" in window) {
-        websocket = new MozWebSocket(url);
-    } else {
-        websocket = new SockJS(url);
-    }
-
-    //连接发生错误的回调方法
-    websocket.onerror = function () {
-        console.log("初始化失败");
-        this.$notify.error({
-            title: "错误",
-            message: "连接失败，请检查网络"
-        });
-    };
-
-    //连接成功建立的回调方法
-    websocket.onopen = function () {
-        console.log("ws初始化成功");
-    };
-
-    //接收到消息的回调方法
-    websocket.onmessage = function (event) {
-        console.log("event", event);
-        //let _data = event.data;
-        let _data = JSON.parse(_data);
-        if (_data.code == 200) {
-            toastr.info(_data.msg);
-        } else {
-            toastr.error(_data.msg);
+    function newWebSocket() {
+        let heartCheck = {
+            timeout: 30000,        // 9分钟发一次心跳，比server端设置的连接时间稍微小一点，在接近断开的情况下以通信的方式去重置连接时间。
+            serverTimeoutObj: null,
+            reset: function () {
+                clearTimeout(this.timeoutObj);
+                clearTimeout(this.serverTimeoutObj);
+                return this;
+            },
+            start: function () {
+                var self = this;
+                this.serverTimeoutObj = setInterval(function () {
+                    if (websocket.readyState == 1) {
+                        console.log("连接状态，发送消息保持连接");
+                        websocket.send("{'cmd':'ping'}");
+                        heartCheck.reset().start();    // 如果获取到消息，说明连接是正常的，重置心跳检测
+                    } else {
+                        console.log("断开状态，尝试重连");
+                        newWebSocket();
+                    }
+                }, this.timeout)
+            }
         }
-    };
 
-    //连接关闭的回调方法
-    websocket.onclose = function () {
-        console.log("初始化失败");
-        this.$notify.error({
-            title: "错误",
-            message: "连接关闭，请刷新浏览器"
-        });
-    };
+        //建立websocket 消息连接
+        let _userId = vc.getData('/nav/getUserInfo').userId;
 
-    //监听窗口关闭事件，当窗口关闭时，主动去关闭websocket连接，防止连接还没断开就关闭窗口，server端会抛异常。
-    window.onbeforeunload = function () {
-        websocket.close();
-    };
+        let _protocol = window.location.protocol;
+        let url = '';
+        if (_protocol.startsWith('https')) {
+            url =
+                "wss://" + window.location.host + "/ws/message/" +
+                _userId;
+        } else {
+            url =
+                "ws://" + window.location.host + "/ws/message/" +
+                _userId;
+            // url =
+            //     "ws://demo.homecommunity.cn/ws/message/" +
+            //     _userId;
+        }
+
+
+        if ("WebSocket" in window) {
+            websocket = new WebSocket(url);
+        } else if ("MozWebSocket" in window) {
+            websocket = new MozWebSocket(url);
+        } else {
+            websocket = new SockJS(url);
+        }
+
+        //连接发生错误的回调方法
+        websocket.onerror = function (_err) {
+            console.log("初始化失败", _err);
+            this.$notify.error({
+                title: "错误",
+                message: "连接失败，请检查网络"
+            });
+        };
+
+        //连接成功建立的回调方法
+        websocket.onopen = function () {
+            heartCheck.reset().start();
+            console.log("ws初始化成功");
+        };
+
+        //接收到消息的回调方法
+        websocket.onmessage = function (event) {
+            heartCheck.reset().start(); 
+            console.log("event", event);
+            let _data = event.data;
+            try {
+                _data = JSON.parse(_data);
+            } catch (err) {
+                return;
+            }
+            
+            if (_data.code == 200) {
+                toastr.info(_data.msg);
+            } else {
+                toastr.error(_data.msg);
+            }
+        };
+
+        //连接关闭的回调方法
+        websocket.onclose = function () {
+            console.log("初始化失败");
+            //newWebSocket();
+            this.$notify.error({
+                title: "错误",
+                message: "连接关闭，请刷新浏览器"
+            });
+        };
+
+        //监听窗口关闭事件，当窗口关闭时，主动去关闭websocket连接，防止连接还没断开就关闭窗口，server端会抛异常。
+        window.onbeforeunload = function () {
+            websocket.close();
+        };
+    }
+
+    newWebSocket();
 
 })(window.vc);