From a41e72f9c2a757ae47b4740d2fd1b4634249b061 Mon Sep 17 00:00:00 2001
From: Pana <pana.wang@outlook.com>
Date: Fri, 8 Aug 2025 15:07:05 +0800
Subject: [PATCH] remove peer from stratum workder if send message failed

---
 crates/stratum/src/lib.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/crates/stratum/src/lib.rs b/crates/stratum/src/lib.rs
index b0f6e515cf..7c8fb2d5b6 100644
--- a/crates/stratum/src/lib.rs
+++ b/crates/stratum/src/lib.rs
@@ -226,8 +226,11 @@ impl StratumImpl {
                         debug!(target: "stratum", "Worker no longer connected: {} addr {}", &worker_id, &addr);
                         hup_peers.insert(**addr);
                     }
-                    Err(e) => {
+                    Err(PushMessageError::Send(e)) => {
                         warn!(target: "stratum", "Unexpected transport error: {:?}", e);
+                        if e.is_disconnected() {
+                            hup_peers.insert(**addr);
+                        }
                     }
                     Ok(_) => {}
                 }