From b236de082a99bd932afe750e82fd163e7163eedd Mon Sep 17 00:00:00 2001
From: Klesti Selimaj <klesti.selimay@icloud.com>
Date: Mon, 23 Feb 2026 05:54:56 +0100
Subject: [PATCH] Fixed ping bug

---
 .../dev/selimaj/session/SessionListener.java  | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/main/java/dev/selimaj/session/SessionListener.java b/src/main/java/dev/selimaj/session/SessionListener.java
index b8638ff..fc7546b 100644
--- a/src/main/java/dev/selimaj/session/SessionListener.java
+++ b/src/main/java/dev/selimaj/session/SessionListener.java
@@ -58,14 +58,33 @@ public class SessionListener implements WebSocket.Listener {
                         new RuntimeException(r.error().toString()));
         }
 
+        ws.request(1);
         return null;
     }
 
+    @Override
+    public void onOpen(WebSocket ws) {
+        ws.request(1);
+    }
+
     @Override
     public CompletionStage<?> onPing(WebSocket ws, ByteBuffer message) {
+        ws.request(1);
         return ws.sendPong(message);
     }
 
+    @Override
+    public CompletionStage<?> onPong(WebSocket ws, ByteBuffer message) {
+        ws.request(1);
+        return null;
+    }
+
+    @Override
+    public CompletionStage<?> onClose(WebSocket ws, int statusCode, String reason) {
+        ws.request(1);
+        return null;
+    }
+
     void send(WebSocket ws, Message msg) throws Exception {
         ws.sendText(mapper.writeValueAsString(msg), true);
     }