mirror of
https://github.com/logos-messaging/logos-delivery.git
synced 2026-06-05 21:49:31 +00:00
75 lines
2.2 KiB
Nim
75 lines
2.2 KiB
Nim
import std/[options, times], chronos
|
|
import brokers/broker_context
|
|
import waku/waku_core, messaging/api/types, waku/api/requests/node
|
|
|
|
type DeliveryState* {.pure.} = enum
|
|
Entry
|
|
SuccessfullyPropagated
|
|
# message is known to be sent to the network but not yet validated
|
|
SuccessfullyValidated
|
|
# message is known to be stored at least on one store node, thus validated
|
|
FallbackRetry # retry sending with fallback processor if available
|
|
NextRoundRetry # try sending in next loop
|
|
FailedToDeliver # final state of failed delivery
|
|
|
|
type DeliveryTask* = ref object
|
|
requestId*: RequestId
|
|
pubsubTopic*: PubsubTopic
|
|
msg*: WakuMessage
|
|
msgHash*: WakuMessageHash
|
|
tryCount*: int
|
|
state*: DeliveryState
|
|
deliveryTime*: Moment
|
|
propagateEventEmitted*: bool
|
|
errorDesc*: string
|
|
|
|
proc new*(
|
|
T: typedesc[DeliveryTask],
|
|
requestId: RequestId,
|
|
envelop: MessageEnvelope,
|
|
brokerCtx: BrokerContext,
|
|
): Result[T, string] =
|
|
let msg = envelop.toWakuMessage()
|
|
# TODO: use sync request for such as soon as available
|
|
let relayShardRes = (
|
|
RequestRelayShard.request(brokerCtx, none[PubsubTopic](), envelop.contentTopic)
|
|
).valueOr:
|
|
error "RequestRelayShard.request failed", error = error
|
|
return err("Failed create DeliveryTask: " & $error)
|
|
|
|
let pubsubTopic = relayShardRes.relayShard.toPubsubTopic()
|
|
let msgHash = computeMessageHash(pubsubTopic, msg)
|
|
|
|
return ok(
|
|
T(
|
|
requestId: requestId,
|
|
pubsubTopic: pubsubTopic,
|
|
msg: msg,
|
|
msgHash: msgHash,
|
|
tryCount: 0,
|
|
state: DeliveryState.Entry,
|
|
)
|
|
)
|
|
|
|
func `==`*(r, l: DeliveryTask): bool =
|
|
if r.isNil() == l.isNil():
|
|
r.isNil() or r.msgHash == l.msgHash
|
|
else:
|
|
false
|
|
|
|
proc messageAge*(self: DeliveryTask): timer.Duration =
|
|
let actual = getNanosecondTime(getTime().toUnixFloat())
|
|
if self.msg.timestamp >= 0 and self.msg.timestamp < actual:
|
|
nanoseconds(actual - self.msg.timestamp)
|
|
else:
|
|
ZeroDuration
|
|
|
|
proc deliveryAge*(self: DeliveryTask): timer.Duration =
|
|
if self.state == DeliveryState.SuccessfullyPropagated:
|
|
timer.Moment.now() - self.deliveryTime
|
|
else:
|
|
ZeroDuration
|
|
|
|
proc isEphemeral*(self: DeliveryTask): bool =
|
|
return self.msg.ephemeral
|