From 4d88bcbce33b6328a20760d24307dd40ba7c5fb0 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Fri, 6 Mar 2026 09:09:42 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E4=BC=98=E5=8C=96=20heartbeat=20?=
 =?UTF-8?q?=E4=B8=AD=20status=20=E9=94=81=E7=B2=92=E5=BA=A6=EF=BC=8C?=
 =?UTF-8?q?=E4=BD=BF=E7=94=A8=20get=5Fsession=20=E5=87=BD=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/connection.rs | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/src/connection.rs b/src/connection.rs
index aea8441..bfde9ca 100644
--- a/src/connection.rs
+++ b/src/connection.rs
@@ -277,10 +277,9 @@ impl ConnectionManager {
                 ticker.tick().await;
 
                 // 检查session是否有效
-                let session_valid = {
-                    let status = manager.status.read().await;
-                    if let Some(conn_status) = status.get(&source_id) {
-                        if let Some(session) = conn_status.session.as_ref() {
+                let session = manager.get_session(source_id).await;
+
+                let session_valid = if let Some(session) = session {
                             // 尝试读取当前时间来验证连接
                             let node_id = NodeId::new(0, 2258); // ServerCurrentTime节点
                             let read_request = ReadValueId {
@@ -294,12 +293,8 @@ impl ConnectionManager {
                                 Ok(_) => true,
                                 Err(_) => false,
                             }
-                        } else {
-                            false
-                        }
-                    } else {
-                        false
-                    }
+                } else {
+                    false
                 };
 
                 if !session_valid {