Validating illegal argument exception when closing

Author: LikeTheSalad

opamp-client/src/main/java/io/opentelemetry/opamp/client/internal/connectivity/websocket/OkHttpWebSocket.java

@@ -52,8 +52,13 @@ public boolean send(byte[] request) {
   @Override
   public void close(int code, @Nullable String reason) {
     if (status.compareAndSet(Status.RUNNING, Status.CLOSING)) {
-      if (!getWebSocket().close(code, reason)) {
-        status.set(Status.NOT_RUNNING);
+      try {
+        if (!getWebSocket().close(code, reason)) {
+          status.set(Status.NOT_RUNNING);
+        }
+      } catch (IllegalArgumentException e) {
+        status.set(Status.RUNNING);
+        throw e;
       }
     }
   }