56 lines
1.9 KiB
Nim
56 lines
1.9 KiB
Nim
# nim-raft
|
|
# Copyright (c) 2023 Status Research & Development GmbH
|
|
# Licensed under either of
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
# at your option.
|
|
# This file may not be copied, modified, or distributed except according to
|
|
# those terms.
|
|
|
|
import basic_timers
|
|
import basic_state_machine
|
|
import std/tables
|
|
|
|
export raft_api
|
|
|
|
type
|
|
BasicRaftNode* = RaftNode[SmCommand, SmState]
|
|
|
|
BasicRaftCluster* = ref object
|
|
nodes*: Table[RaftNodeId, BasicRaftNode]
|
|
|
|
proc BasicRaftClusterRaftMessageSendCallbackCreate[SmCommandType, SmStateType](cluster: BasicRaftCluster): RaftMessageSendCallback[SmCommandType, SmStateType] =
|
|
proc (msg: RaftMessageBase[SmCommandType, SmStateType]): Future[RaftMessageResponseBase[SmCommandType, SmStateType]] {.async, gcsafe.} =
|
|
result = await cluster.nodes[msg.receiverId].RaftNodeMessageDeliver(msg)
|
|
|
|
proc BasicRaftClusterStart*(cluster: BasicRaftCluster) =
|
|
for id, node in cluster.nodes:
|
|
RaftNodeStart(node)
|
|
|
|
proc BasicRaftClusterGetLeaderId*(cluster: BasicRaftCluster): UUID =
|
|
result = DefaultUUID
|
|
for id, node in cluster.nodes:
|
|
if RaftNodeIsLeader(node):
|
|
return RaftNodeIdGet(node)
|
|
|
|
proc BasicRaftClusterClientRequest*(cluster: BasicRaftCluster, req: RaftNodeClientRequest): Future[RaftNodeClientResponse] {.async.} =
|
|
case req.op:
|
|
of rncroRequestSmState:
|
|
var
|
|
nodeId = cluster.nodesIds[BasicRaftClusterGetLeaderId(cluster)]
|
|
|
|
result = await cluster.nodes[nodeId].RaftNodeServeClientRequest(req)
|
|
|
|
of rncroExecSmCommand:
|
|
discard
|
|
|
|
proc BasicRaftClusterInit*(nodesIds: seq[RaftNodeId]): BasicRaftCluster =
|
|
new(result)
|
|
for nodeId in nodesIds:
|
|
var
|
|
peersIds = nodesIds
|
|
|
|
peersIds.del(peersIds.find(nodeId))
|
|
result.nodes[nodeId] = BasicRaftNode.new(nodeId, peersIds, BasicRaftClusterRaftMessageSendCallbackCreate[SmCommand, SmState](result), electionTimeout=50, heartBeatTimeout=50)
|
|
|