Fixing bug in nodeMap update logic

maxpert · Sep 1, 2022 · 19c60fa · 19c60fa
1 parent 6ed4a3b
commit 19c60fa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/raft.go b/lib/raft.go
@@ -264,10 +264,10 @@ func (r *RaftServer) mutateNodeMap(nodeID uint64, f func(map[uint64]bool)) {
     }
 
     f(m)
+    r.nodeMap[nodeID] = m
+
     if len(m) <= 0 {
         delete(r.nodeMap, nodeID)
-    } else {
-        r.nodeMap[nodeID] = m
     }
 }