mirror of https://github.com/waku-org/nwaku.git
91 lines
2.4 KiB
Nim
91 lines
2.4 KiB
Nim
when (NimMajor, NimMinor) < (1, 4):
|
|
{.push raises: [Defect].}
|
|
else:
|
|
{.push raises: [].}
|
|
|
|
import std/[options], chronos
|
|
|
|
import waku_node, ../waku_rln_relay
|
|
|
|
type
|
|
HealthStatus* = enum
|
|
INITIALIZING
|
|
SYNCHRONIZING
|
|
READY
|
|
NOT_READY
|
|
NOT_MOUNTED
|
|
SHUTTING_DOWN
|
|
|
|
ProtocolHealth* = object
|
|
protocol*: string
|
|
health*: HealthStatus
|
|
|
|
HealthReport* = object
|
|
nodeHealth*: HealthStatus
|
|
protocolsHealth*: seq[ProtocolHealth]
|
|
|
|
WakuNodeHealthMonitor* = ref object
|
|
nodeHealth: HealthStatus
|
|
node: Option[WakuNode]
|
|
|
|
proc `$`*(t: HealthStatus): string =
|
|
result =
|
|
case t
|
|
of INITIALIZING: "Initializing"
|
|
of SYNCHRONIZING: "Synchronizing"
|
|
of READY: "Ready"
|
|
of NOT_READY: "Not Ready"
|
|
of NOT_MOUNTED: "Not Mounted"
|
|
of SHUTTING_DOWN: "Shutting Down"
|
|
|
|
proc init*(
|
|
t: typedesc[HealthStatus], strRep: string
|
|
): HealthStatus {.raises: [ValueError].} =
|
|
case strRep
|
|
of "Initializing":
|
|
return HealthStatus.INITIALIZING
|
|
of "Synchronizing":
|
|
return HealthStatus.SYNCHRONIZING
|
|
of "Ready":
|
|
return HealthStatus.READY
|
|
of "Not Ready":
|
|
return HealthStatus.NOT_READY
|
|
of "Not Mounted":
|
|
return HealthStatus.NOT_MOUNTED
|
|
of "Shutting Down":
|
|
return HealthStatus.SHUTTING_DOWN
|
|
else:
|
|
raise newException(ValueError, "Invalid HealthStatus string representation")
|
|
|
|
const FutIsReadyTimout = 5.seconds
|
|
|
|
proc getNodeHealthReport*(hm: WakuNodeHealthMonitor): Future[HealthReport] {.async.} =
|
|
result.nodeHealth = hm.nodeHealth
|
|
|
|
if hm.node.isSome() and hm.node.get().wakuRlnRelay != nil:
|
|
let getRlnRelayHealth = proc(): Future[HealthStatus] {.async.} =
|
|
let isReadyStateFut = hm.node.get().wakuRlnRelay.isReady()
|
|
if not await isReadyStateFut.withTimeout(FutIsReadyTimout):
|
|
return HealthStatus.NOT_READY
|
|
|
|
try:
|
|
if not isReadyStateFut.completed():
|
|
return HealthStatus.NOT_READY
|
|
elif isReadyStateFut.read():
|
|
return HealthStatus.READY
|
|
|
|
return HealthStatus.SYNCHRONIZING
|
|
except:
|
|
error "exception reading state: " & getCurrentExceptionMsg()
|
|
return HealthStatus.NOT_READY
|
|
|
|
result.protocolsHealth.add(
|
|
ProtocolHealth(protocol: "Rln Relay", health: await getRlnRelayHealth())
|
|
)
|
|
|
|
proc setNode*(hm: WakuNodeHealthMonitor, node: WakuNode) =
|
|
hm.node = some(node)
|
|
|
|
proc setOverallHealth*(hm: WakuNodeHealthMonitor, health: HealthStatus) =
|
|
hm.nodeHealth = health
|