Browse Source

merge https://github.com/mayswind/AriaNg/commit/317348ebbdab493703ddc7f7c5273aa05bdc0460

MaysWind 6 years ago
parent
commit
1c59573dac

+ 1 - 0
app/scripts/config/constants.js

@@ -13,6 +13,7 @@
         defaultHost: 'localhost',
         defaultSecureProtocol: 'https',
         defaultPathSeparator: '/',
+        websocketAutoReconnect: true,
         globalStatStorageCapacity: 120,
         taskStatStorageCapacity: 300,
         lazySaveTimeout: 500,

+ 4 - 2
app/scripts/services/aria2WebSocketRpcService.js

@@ -1,7 +1,7 @@
 (function () {
     'use strict';
 
-    angular.module('ariaNg').factory('aria2WebSocketRpcService', ['$q', '$websocket', 'ariaNgSettingService', 'ariaNgLogService', function ($q, $websocket, ariaNgSettingService, ariaNgLogService) {
+    angular.module('ariaNg').factory('aria2WebSocketRpcService', ['$q', '$websocket', 'ariaNgConstants', 'ariaNgSettingService', 'ariaNgLogService', function ($q, $websocket, ariaNgConstants, ariaNgSettingService, ariaNgLogService) {
         var rpcUrl = ariaNgSettingService.getCurrentRpcUrl();
         var socketClient = null;
 
@@ -72,7 +72,9 @@
         var getSocketClient = function (context) {
             if (socketClient === null) {
                 try {
-                    socketClient = $websocket(rpcUrl);
+                    socketClient = $websocket(rpcUrl, {
+                        reconnectIfNotNormalClose: ariaNgConstants.websocketAutoReconnect
+                    });
 
                     socketClient.onMessage(function (message) {
                         if (!message || !message.data) {