mirror of
https://github.com/logos-messaging/logos-messaging-nim.git
synced 2026-02-22 14:53:27 +00:00
82 lines
2.8 KiB
Nim
82 lines
2.8 KiB
Nim
|
import chronicles, chronos, results
|
||
|
|
import std/options
|
||
|
|
|
||
|
|
import
|
||
|
|
waku/node/peer_manager,
|
||
|
|
waku/waku_core,
|
||
|
|
waku/waku_lightpush/[common, client, rpc],
|
||
|
|
waku/common/broker/broker_context
|
||
|
|
|
||
|
|
import ./[delivery_task, send_processor]
|
||
|
|
|
||
|
|
logScope:
|
||
|
|
topics = "send service lightpush processor"
|
||
|
|
|
||
|
|
type LightpushSendProcessor* = ref object of BaseSendProcessor
|
||
|
|
peerManager: PeerManager
|
||
|
|
lightpushClient: WakuLightPushClient
|
||
|
|
|
||
|
|
proc new*(
|
||
|
|
T: typedesc[LightpushSendProcessor],
|
||
|
|
peerManager: PeerManager,
|
||
|
|
lightpushClient: WakuLightPushClient,
|
||
|
|
brokerCtx: BrokerContext,
|
||
|
|
): T =
|
||
|
|
return
|
||
|
|
T(peerManager: peerManager, lightpushClient: lightpushClient, brokerCtx: brokerCtx)
|
||
|
|
|
||
|
|
proc isLightpushPeerAvailable(
|
||
|
|
self: LightpushSendProcessor, pubsubTopic: PubsubTopic
|
||
|
|
): bool =
|
||
|
|
return self.peerManager.selectPeer(WakuLightPushCodec, some(pubsubTopic)).isSome()
|
||
|
|
|
||
|
|
method isValidProcessor*(
|
||
|
|
self: LightpushSendProcessor, task: DeliveryTask
|
||
|
|
): bool {.gcsafe.} =
|
||
|
|
return self.isLightpushPeerAvailable(task.pubsubTopic)
|
||
|
|
|
||
|
|
method sendImpl*(
|
||
|
|
self: LightpushSendProcessor, task: DeliveryTask
|
||
|
|
): Future[void] {.async.} =
|
||
|
|
task.tryCount.inc()
|
||
|
|
info "Trying message delivery via Lightpush",
|
||
|
|
requestId = task.requestId,
|
||
|
|
msgHash = task.msgHash.to0xHex(),
|
||
|
|
tryCount = task.tryCount
|
||
|
|
|
||
|
|
let peer = self.peerManager.selectPeer(WakuLightPushCodec, some(task.pubsubTopic)).valueOr:
|
||
|
|
debug "No peer available for Lightpush, request pushed back for next round",
|
||
|
|
requestId = task.requestId
|
||
|
|
task.state = DeliveryState.NextRoundRetry
|
||
|
|
return
|
||
|
|
|
||
|
|
let numLightpushServers = (
|
||
|
|
await self.lightpushClient.publish(some(task.pubsubTopic), task.msg, peer)
|
||
|
|
).valueOr:
|
||
|
|
error "LightpushSendProcessor.sendImpl failed", error = error.desc.get($error.code)
|
||
|
|
case error.code
|
||
|
|
of LightPushErrorCode.NO_PEERS_TO_RELAY, LightPushErrorCode.TOO_MANY_REQUESTS,
|
||
|
|
LightPushErrorCode.OUT_OF_RLN_PROOF, LightPushErrorCode.SERVICE_NOT_AVAILABLE,
|
||
|
|
LightPushErrorCode.INTERNAL_SERVER_ERROR:
|
||
|
|
task.state = DeliveryState.NextRoundRetry
|
||
|
|
else:
|
||
|
|
# the message is malformed, send error
|
||
|
|
task.state = DeliveryState.FailedToDeliver
|
||
|
|
task.errorDesc = error.desc.get($error.code)
|
||
|
|
task.deliveryTime = Moment.now()
|
||
|
|
return
|
||
|
|
|
||
|
|
if numLightpushServers > 0:
|
||
|
|
info "Message propagated via Lightpush",
|
||
|
|
requestId = task.requestId, msgHash = task.msgHash.to0xHex()
|
||
|
|
task.state = DeliveryState.SuccessfullyPropagated
|
||
|
|
task.deliveryTime = Moment.now()
|
||
|
|
# TODO: with a simple retry processor it might be more accurate to say `Sent`
|
||
|
|
else:
|
||
|
|
# Controversial state, publish says ok but no peer. It should not happen.
|
||
|
|
debug "Lightpush publish returned zero peers, request pushed back for next round",
|
||
|
|
requestId = task.requestId
|
||
|
|
task.state = DeliveryState.NextRoundRetry
|
||
|
|
|
||
|
|
return
|