logos-messaging/logos-messaging-nim
1
0
Fork 0
mirror of https://github.com/logos-messaging/logos-messaging-nim.git synced 2026-03-19 19:13:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

add optional reputation manager to Lightpush client initialization

Browse Source
This commit is contained in:
Sergei Tikhomirov 2025-02-27 16:42:23 +01:00
parent 64498403d3
commit ef67d542ab
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/waku_lightpush/lightpush_utils.nim
View File

@ -27,5 +27,7 @@ proc newTestWakuLightpushNode*(
proc newTestWakuLightpushClient*(switch: Switch): WakuLightPushClient =
let peerManager = PeerManager.new(switch)
let reputationManager = ReputationManager.new()
WakuLightPushClient.new(peerManager, reputationManager, rng)
var reputationManager = none(ReputationManager)
when defined(reputation):
reputationManager = some(ReputationManager.new())
WakuLightPushClient.new(peerManager, rng, reputationManager)

11
waku/waku_lightpush/client.nim
View File

@ -26,12 +26,15 @@ type WakuLightPushClient* = ref object
proc new*(
T: type WakuLightPushClient,
peerManager: PeerManager,
reputationManager: ReputationManager,
rng: ref rand.HmacDrbgContext,
reputationManager: Option[ReputationManager] = none(ReputationManager),
): T =
WakuLightPushClient(
peerManager: peerManager, reputationManager: reputationManager, rng: rng
)
if reputationManager.isSome:
WakuLightPushClient(
peerManager: peerManager, rng: rng, reputationManager: reputationManager.get()
)
else:
WakuLightPushClient(peerManager: peerManager, rng: rng)
proc addPublishObserver*(wl: WakuLightPushClient, obs: PublishObserver) =
wl.publishObservers.add(obs)