forked from zhurui/management
52 lines
1.5 KiB
JavaScript
52 lines
1.5 KiB
JavaScript
|
'use strict';
|
||
|
|
/* global __webpack_dev_server_client__ */
|
||
|
|
|
||
|
|
/* eslint-disable
|
||
|
|
camelcase
|
||
|
|
*/
|
||
|
|
// this SockJSClient is here as a default fallback, in case inline mode
|
||
|
|
// is off or the client is not injected. This will be switched to
|
||
|
|
// WebsocketClient when it becomes the default
|
||
|
|
// important: the path to SockJSClient here is made to work in the 'client'
|
||
|
|
// directory, but is updated via the webpack compilation when compiled from
|
||
|
|
// the 'client-src' directory
|
||
|
|
|
||
|
|
var Client = typeof __webpack_dev_server_client__ !== 'undefined' ? __webpack_dev_server_client__ : // eslint-disable-next-line import/no-unresolved
|
||
|
|
require('./clients/SockJSClient');
|
||
|
|
var retries = 0;
|
||
|
|
var client = null;
|
||
|
|
|
||
|
|
var socket = function initSocket(url, handlers) {
|
||
|
|
client = new Client(url);
|
||
|
|
client.onOpen(function () {
|
||
|
|
retries = 0;
|
||
|
|
});
|
||
|
|
client.onClose(function () {
|
||
|
|
if (retries === 0) {
|
||
|
|
handlers.close();
|
||
|
|
} // Try to reconnect.
|
||
|
|
|
||
|
|
|
||
|
|
client = null; // After 10 retries stop trying, to prevent logspam.
|
||
|
|
|
||
|
|
if (retries <= 10) {
|
||
|
|
// Exponentially increase timeout to reconnect.
|
||
|
|
// Respectfully copied from the package `got`.
|
||
|
|
// eslint-disable-next-line no-mixed-operators, no-restricted-properties
|
||
|
|
var retryInMs = 1000 * Math.pow(2, retries) + Math.random() * 100;
|
||
|
|
retries += 1;
|
||
|
|
setTimeout(function () {
|
||
|
|
socket(url, handlers);
|
||
|
|
}, retryInMs);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
client.onMessage(function (data) {
|
||
|
|
var msg = JSON.parse(data);
|
||
|
|
|
||
|
|
if (handlers[msg.type]) {
|
||
|
|
handlers[msg.type](msg.data);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
module.exports = socket;
|