heartbeat: Improve locking

akiradeveloper · May 25, 2024 · f4a1a72 · f4a1a72
1 parent 5c38619
commit f4a1a72
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lolraft/src/communicator/heartbeat_multiplex.rs b/lolraft/src/communicator/heartbeat_multiplex.rs
@@ -30,10 +30,13 @@ pub async fn run(
     loop {
         tokio::time::sleep(Duration::from_millis(300)).await;
 
-        let states = {
+        let heartbeats = {
             let mut buf = buf.lock();
-            let heartbeats = buf.drain();
+            let out = buf.drain();
+            out
+        };
 
+        let states = {
             let mut out = HashMap::new();
             for (lane_id, heartbeat) in heartbeats {
                 let state = raft::LeaderCommitState {