status-im
/
nim-raft
mirror of https://github.com/status-im/nim-raft.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix

This commit is contained in:
Raycho Mukelov 2023-10-20 01:36:40 +03:00
parent 14d49d5737
commit dfaf67f9ef
3 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
raft/consensus_module.nim
View File

@ -172,3 +172,7 @@ proc raftNodeReplicateSmCommand*[SmCommandType, SmStateType](node: RaftNode[SmCo
) )
node.replicateFuts.add(node.msgSendCallback(msg)) node.replicateFuts.add(node.msgSendCallback(msg))
node.commitIndex.inc
raftNodeApplyLogEntry(node, raftNodeLogEntryGet(node, node.commitIndex)) # Apply to state machine

8
tests/basic_cluster.nim
View File

@ -18,6 +18,7 @@ type
BasicRaftCluster* = ref object BasicRaftCluster* = ref object
nodes*: Table[RaftNodeId, BasicRaftNode] nodes*: Table[RaftNodeId, BasicRaftNode]
nodesLock*: RLock
proc basicRaftClusterRaftMessageSendCallbackCreate[SmCommandType, SmStateType](cluster: BasicRaftCluster): RaftMessageSendCallback[SmCommandType, SmStateType] = proc basicRaftClusterRaftMessageSendCallbackCreate[SmCommandType, SmStateType](cluster: BasicRaftCluster): RaftMessageSendCallback[SmCommandType, SmStateType] =
proc (msg: RaftMessageBase[SmCommandType, SmStateType]): Future[RaftMessageResponseBase[SmCommandType, SmStateType]] {.async, gcsafe.} = proc (msg: RaftMessageBase[SmCommandType, SmStateType]): Future[RaftMessageResponseBase[SmCommandType, SmStateType]] {.async, gcsafe.} =
@ -29,9 +30,10 @@ proc basicRaftClusterStart*(cluster: BasicRaftCluster) =
proc basicRaftClusterGetLeaderId*(cluster: BasicRaftCluster): UUID = proc basicRaftClusterGetLeaderId*(cluster: BasicRaftCluster): UUID =
result = DefaultUUID result = DefaultUUID
for id, node in cluster.nodes: withRLock(cluster.nodesLock):
if raftNodeIsLeader(node): for id, node in cluster.nodes:
return raftNodeIdGet(node) if raftNodeIsLeader(node):
return raftNodeIdGet(node)
proc basicRaftClusterClientRequest*(cluster: BasicRaftCluster, req: RaftNodeClientRequest): Future[RaftNodeClientResponse] {.async.} = proc basicRaftClusterClientRequest*(cluster: BasicRaftCluster, req: RaftNodeClientRequest): Future[RaftNodeClientResponse] {.async.} =
case req.op: case req.op:

2
tests/test_basic_cluster_election.nim
View File

@ -20,7 +20,7 @@ proc basicClusterElectionMain*() =
test "Basic Raft Cluster Init (5 nodes)": test "Basic Raft Cluster Init (5 nodes)":
for i in 0..4: for i in 0..4:
nodesIds[i] = genUUID() nodesIds[i] = genUUID()
cluster = basicRaftClusterInit(nodesIds, 5, 5, 20, 1) cluster = basicRaftClusterInit(nodesIds, 150, 150, 20, 20)
check cluster != nil check cluster != nil
test "Start Basic Raft Cluster and wait it to converge for a 2 seconds interval (Elect a Leader)": test "Start Basic Raft Cluster and wait it to converge for a 2 seconds interval (Elect a Leader)":