From a3c3aab44e268a972113eec6e03d6a3ed874f4f6 Mon Sep 17 00:00:00 2001 From: Prem Chaitanya Prathi Date: Wed, 25 Oct 2023 02:11:42 +0530 Subject: [PATCH] fix: propagate relay subscribe opt properly (#829) --- waku/v2/protocol/relay/broadcast.go | 6 ++++++ waku/v2/protocol/relay/waku_relay.go | 3 ++- 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/waku/v2/protocol/relay/broadcast.go b/waku/v2/protocol/relay/broadcast.go index dccc8750..36ca3e1e 100644 --- a/waku/v2/protocol/relay/broadcast.go +++ b/waku/v2/protocol/relay/broadcast.go @@ -24,6 +24,12 @@ func DontConsume() BroadcasterOption { } } +func WithConsumerOption(dontConsume bool) BroadcasterOption { + return func(params *BroadcasterParameters) { + params.dontConsume = dontConsume + } +} + // WithBufferSize option let's a user set channel buffer to be set. func WithBufferSize(size int) BroadcasterOption { return func(params *BroadcasterParameters) { diff --git a/waku/v2/protocol/relay/waku_relay.go b/waku/v2/protocol/relay/waku_relay.go index ae69fae8..465f8268 100644 --- a/waku/v2/protocol/relay/waku_relay.go +++ b/waku/v2/protocol/relay/waku_relay.go @@ -339,7 +339,8 @@ func (w *WakuRelay) subscribe(ctx context.Context, contentFilter waku_proto.Cont } } - subscription := w.bcaster.Register(cFilter, WithBufferSize(DefaultRelaySubscriptionBufferSize)) + subscription := w.bcaster.Register(cFilter, WithBufferSize(DefaultRelaySubscriptionBufferSize), + WithConsumerOption(params.dontConsume)) // Create Content subscription w.topicsMutex.RLock()