Config findProviderTimeout through client

ipfs · Nov 22, 2024 · 7409ebe · 7409ebe
1 parent 84f8f13
commit 7409ebe
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/bitswap/client/client.go b/bitswap/client/client.go
@@ -76,6 +76,7 @@ func New(parent context.Context, network bsnet.BitSwapNetwork, bstore blockstore
 	bpm := bsbpm.New()
 	pm := bspm.New(ctx, peerQueueFactory, network.Self())
 	pqm := bspqm.New(ctx, network,
+		bspqm.WithFindProviderTimeout(opts.findProviderTimeout),
 		bspqm.WithMaxConcurrentFinds(opts.maxConcurrentFinds),
 		bspqm.WithMaxProvidersPerFind(opts.maxProvidersPerFind))
 

diff --git a/bitswap/client/optios.go b/bitswap/client/optios.go
@@ -17,6 +17,7 @@ type clientConfig struct {
 	tracer                     tracer.Tracer
 
 	// ProviderQueryManager options.
+	findProviderTimeout time.Duration
 	maxConcurrentFinds  int
 	maxProvidersPerFind int
 }
@@ -92,6 +93,12 @@ func WithoutDuplicatedBlockStats() Option {
 	}
 }
 
+func WithFindProviderTimeout(to time.Duration) Option {
+	return func(c *clientConfig) {
+		c.findProviderTimeout = to
+	}
+}
+
 func WithMaxConcurrentFinds(n int) Option {
 	return func(c *clientConfig) {
 		c.maxConcurrentFinds = n