From 0b5ad1d78ac2480fdce9c826cd3706a8a61a80e3 Mon Sep 17 00:00:00 2001 From: Gabriel mermelstein Date: Fri, 13 Dec 2024 17:25:40 +0100 Subject: [PATCH] requiring user to pass context --- waku/nwaku.go | 4 +--- waku/nwaku_test.go | 4 +++- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/waku/nwaku.go b/waku/nwaku.go index 5f284d8..b6ab7d7 100644 --- a/waku/nwaku.go +++ b/waku/nwaku.go @@ -425,9 +425,7 @@ func (w *Waku) DialPeer(address multiaddr.Multiaddr) error { } // TODO: change pubsub topic to shard notation everywhere -func (w *Waku) RelayPublish(message *pb.WakuMessage, pubsubTopic string) (pb.MessageHash, error) { - ctx, cancel := context.WithTimeout(w.ctx, requestTimeout) - defer cancel() +func (w *Waku) RelayPublish(ctx context.Context, message *pb.WakuMessage, pubsubTopic string) (pb.MessageHash, error) { return w.node.RelayPublish(ctx, message, pubsubTopic) } diff --git a/waku/nwaku_test.go b/waku/nwaku_test.go index 4dd5f28..1301fdb 100644 --- a/waku/nwaku_test.go +++ b/waku/nwaku_test.go @@ -457,7 +457,9 @@ func TestRelay(t *testing.T) { } // send message pubsubTopic := FormatWakuRelayTopic(senderNodeWakuConfig.ClusterID, senderNodeWakuConfig.Shards[0]) - senderNode.RelayPublish(message, pubsubTopic) + ctx, cancel := context.WithTimeout(context.Background(), requestTimeout) + defer cancel() + senderNode.RelayPublish(ctx, message, pubsubTopic) wg := sync.WaitGroup{} wg.Add(1)