socket.js 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. /* global __webpack_dev_server_client__ */
  2. import WebSocketClient from "./clients/WebSocketClient.js";
  3. import { log } from "./utils/log.js";
  4. // this WebsocketClient is here as a default fallback, in case the client is not injected
  5. /* eslint-disable camelcase */
  6. var Client =
  7. // eslint-disable-next-line no-nested-ternary
  8. typeof __webpack_dev_server_client__ !== "undefined" ? typeof __webpack_dev_server_client__.default !== "undefined" ? __webpack_dev_server_client__.default : __webpack_dev_server_client__ : WebSocketClient;
  9. /* eslint-enable camelcase */
  10. var retries = 0;
  11. var maxRetries = 10;
  12. // Initialized client is exported so external consumers can utilize the same instance
  13. // It is mutable to enforce singleton
  14. // eslint-disable-next-line import/no-mutable-exports
  15. export var client = null;
  16. /**
  17. * @param {string} url
  18. * @param {{ [handler: string]: (data?: any, params?: any) => any }} handlers
  19. * @param {number} [reconnect]
  20. */
  21. var socket = function initSocket(url, handlers, reconnect) {
  22. client = new Client(url);
  23. client.onOpen(function () {
  24. retries = 0;
  25. if (typeof reconnect !== "undefined") {
  26. maxRetries = reconnect;
  27. }
  28. });
  29. client.onClose(function () {
  30. if (retries === 0) {
  31. handlers.close();
  32. }
  33. // Try to reconnect.
  34. client = null;
  35. // After 10 retries stop trying, to prevent logspam.
  36. if (retries < maxRetries) {
  37. // Exponentially increase timeout to reconnect.
  38. // Respectfully copied from the package `got`.
  39. // eslint-disable-next-line no-restricted-properties
  40. var retryInMs = 1000 * Math.pow(2, retries) + Math.random() * 100;
  41. retries += 1;
  42. log.info("Trying to reconnect...");
  43. setTimeout(function () {
  44. socket(url, handlers, reconnect);
  45. }, retryInMs);
  46. }
  47. });
  48. client.onMessage(
  49. /**
  50. * @param {any} data
  51. */
  52. function (data) {
  53. var message = JSON.parse(data);
  54. if (handlers[message.type]) {
  55. handlers[message.type](message.data, message.params);
  56. }
  57. });
  58. };
  59. export default socket;