feat: update same term handling

Bisonai · Sep 26, 2024 · 57128db · 57128db
1 parent c127b90
commit 57128db
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/node/pkg/raft/raft.go b/node/pkg/raft/raft.go
@@ -150,6 +150,17 @@ func (r *Raft) handleHeartbeat(msg Message) error {
 		r.LeaderID = heartbeatMessage.LeaderID
 
 		return nil
+	} else if heartbeatMessage.Term == currentTerm {
+		if currentRole == Leader {
+			if r.GetHostId() < heartbeatMessage.LeaderID {
+				r.ResignLeader()
+				r.LeaderID = heartbeatMessage.LeaderID
+			} else {
+				return nil
+			}
+		} else {
+			r.LeaderID = heartbeatMessage.LeaderID
+		}
 	}
 
 	return nil