diff --git a/src/scripts/config/constants.js b/src/scripts/config/constants.js index 580ab20..c27b03c 100644 --- a/src/scripts/config/constants.js +++ b/src/scripts/config/constants.js @@ -13,6 +13,7 @@ defaultHost: 'localhost', defaultSecureProtocol: 'https', defaultPathSeparator: '/', + websocketAutoReconnect: true, globalStatStorageCapacity: 120, taskStatStorageCapacity: 300, lazySaveTimeout: 500, diff --git a/src/scripts/services/aria2WebSocketRpcService.js b/src/scripts/services/aria2WebSocketRpcService.js index 29cfb22..16cb19c 100644 --- a/src/scripts/services/aria2WebSocketRpcService.js +++ b/src/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) {