From 6069c38a13d9f644db5c448efad0e021f3e0e3b1 Mon Sep 17 00:00:00 2001
From: ylz <yuanlinzhu1995@gmail.com>
Date: Fri, 15 Dec 2023 16:42:59 +0800
Subject: [PATCH] fix npe when enable the testWhileIdle parameter

fix
---
 .../mysql/nio/handler/ConnectionHeartBeatHandler.java        | 5 +++--
 1 file changed, 3 insertions(+), 2 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
index ede5ab5548..162769e7f5 100644
--- 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()) {
+            listener.onHeartbeatSuccess((PooledConnection) service.getConnection());
+        }
     }
 
     @Override