fix: process_kern_request needs guard to avoid looping

akiradeveloper · Nov 9, 2024 · 2701967 · 2701967
1 parent e6a3c2a
commit 2701967
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sorock/src/process/raft_process/responder.rs b/sorock/src/process/raft_process/responder.rs
@@ -46,6 +46,8 @@ impl RaftProcess {
 
             rx.await?;
         } else {
+            // Avoid looping.
+            ensure!(self.driver.self_node_id() != leader_id);
             let conn = self.driver.connect(leader_id);
             conn.process_kern_request(req).await?;
         }
@@ -80,7 +82,7 @@ impl RaftProcess {
 
             rx.await?
         } else {
-            // This check is to avoid looping.
+            // Avoid looping.
             ensure!(self.driver.self_node_id() != leader_id);
             let conn = self.driver.connect(leader_id);
             conn.process_user_read_request(req).await?
@@ -116,9 +118,8 @@ impl RaftProcess {
 
             rx.await?
         } else {
-            // This check is to avoid looping.
+            // Avoid looping.
             ensure!(self.driver.self_node_id() != leader_id);
-
             let conn = self.driver.connect(leader_id);
             conn.process_user_write_request(req).await?
         };