fix websocket cleanup

Author: CollinBeczak

src/services/Server/WebSocketClient.js

@@ -153,12 +153,41 @@ export default class WebSocketClient {
     this.sendMessage(pingMessage, true);
   }
 
+  /**
+   * Cleanup websocket connection and intervals
+   */
+  cleanup() {
+    if (this.websocket) {
+      this.websocket.close();
+      this.websocket = null;
+    }
+
+    if (this.pingHandle) {
+      clearInterval(this.pingHandle);
+      this.pingHandle = null;
+    }
+
+    if (this.reconnectionHandle) {
+      clearTimeout(this.reconnectionHandle);
+      this.reconnectionHandle = null;
+    }
+
+    this.reconnectionAttempts = 0;
+    this.queuedMessages = [];
+  }
+
   /**
    * Handles websocket close events, attempting to reconnect
    *
    * @private
    */
   handleClose() {
+    // Clear ping interval on close
+    if (this.pingHandle) {
+      clearInterval(this.pingHandle);
+      this.pingHandle = null;
+    }
+
     this.connect();
   }