fix npe when enable the testWhileIdle parameter

actiontech · Dec 15, 2023 · 5f2b5da · 5f2b5da
1 parent 26dd566
commit 5f2b5da
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main/java/com/actiontech/dble/backend/mysql/nio/handler/ConnectionHeartBeatHandler.java b/src/main/java/com/actiontech/dble/backend/mysql/nio/handler/ConnectionHeartBeatHandler.java
@@ -88,9 +88,10 @@ public void okResponse(byte[] ok, AbstractService service) {
             }
             return;
         }
-
-        heartbeatTimeout.cancel();
-        listener.onHeartbeatSuccess((PooledConnection) service.getConnection());
+        if (!heartbeatTimeout.isExpired()) {
+            heartbeatTimeout.cancel();
+            listener.onHeartbeatSuccess((PooledConnection) service.getConnection());
+        }
     }
 
     @Override