From de0b6332b7cb425abb4194ce3abd732e35848f1a Mon Sep 17 00:00:00 2001
From: Alex <alex@roese.dev>
Date: Sun, 13 Mar 2022 20:18:18 +0100
Subject: [PATCH] added WebsocketNotConnectedException handling

---
 src/main/java/earth/krakatao/SocketClient.java | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/main/java/earth/krakatao/SocketClient.java b/src/main/java/earth/krakatao/SocketClient.java
index a77bb7f..a0b00e4 100644
--- a/src/main/java/earth/krakatao/SocketClient.java
+++ b/src/main/java/earth/krakatao/SocketClient.java
@@ -11,6 +11,7 @@ import java.util.Timer;
 import java.util.TimerTask;
 import lombok.Getter;
 import org.java_websocket.client.WebSocketClient;
+import org.java_websocket.exceptions.WebsocketNotConnectedException;
 import org.java_websocket.handshake.ServerHandshake;
 
 public class SocketClient extends WebSocketClient {
@@ -32,12 +33,17 @@ public class SocketClient extends WebSocketClient {
   }
 
   public void SendMessage(KraSocketClientProtocolMessage kraProtocolMessage) {
-    if (isOpen()) {
-      if (kraProtocolMessage.getStatus() == KraSocketClientProtocolStatus.GET.getStatus()) {
-        this.kraProtocol.getSendQueueMessages()
-            .put(kraProtocolMessage.getCmdID(), kraProtocolMessage);
+    try {
+      if (isOpen()) {
+        if (kraProtocolMessage.getStatus() == KraSocketClientProtocolStatus.GET.getStatus()) {
+          this.kraProtocol.getSendQueueMessages()
+              .put(kraProtocolMessage.getCmdID(), kraProtocolMessage);
+        }
+        send(this.kraProtocol.EncodeMessage(kraProtocolMessage));
       }
-      send(this.kraProtocol.EncodeMessage(kraProtocolMessage));
+    } catch (WebsocketNotConnectedException exception) {
+      KraSocketClient.getLogger().info("SendMessage error");
+      exception.printStackTrace();
     }
   }