feat: implement raft election and term state

This commit is contained in:
Joseph Doherty
2026-02-23 06:11:28 -05:00
parent f1d3c19594
commit 66ec378bdc
4 changed files with 125 additions and 0 deletions

View File

@@ -0,0 +1,56 @@
namespace NATS.Server.Raft;
public sealed class RaftNode
{
private int _votesReceived;
public string Id { get; }
public int Term => TermState.CurrentTerm;
public RaftRole Role { get; private set; } = RaftRole.Follower;
public RaftTermState TermState { get; } = new();
public long AppliedIndex { get; set; }
public RaftNode(string id)
{
Id = id;
}
public void StartElection(int clusterSize)
{
Role = RaftRole.Candidate;
TermState.CurrentTerm++;
TermState.VotedFor = Id;
_votesReceived = 1;
TryBecomeLeader(clusterSize);
}
public VoteResponse GrantVote(int term)
{
if (term < TermState.CurrentTerm)
return new VoteResponse { Granted = false };
TermState.CurrentTerm = term;
return new VoteResponse { Granted = true };
}
public void ReceiveVote(VoteResponse response, int clusterSize = 3)
{
if (!response.Granted)
return;
_votesReceived++;
TryBecomeLeader(clusterSize);
}
private void TryBecomeLeader(int clusterSize)
{
var quorum = (clusterSize / 2) + 1;
if (_votesReceived >= quorum)
Role = RaftRole.Leader;
}
public void RequestStepDown()
{
Role = RaftRole.Follower;
}
}