gateway: implement passthroug WithContextHint on ipfsBackendWithMetrics

ipfs · Jan 12, 2024 · 7b2296c · 7b2296c
1 parent bcb321c
commit 7b2296c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/gateway/metrics.go b/gateway/metrics.go
@@ -179,6 +179,14 @@ func (b *ipfsBackendWithMetrics) GetDNSLinkRecord(ctx context.Context, fqdn stri
 }
 
 var _ IPFSBackend = (*ipfsBackendWithMetrics)(nil)
+var _ WithContextHint = (*ipfsBackendWithMetrics)(nil)
+
+func (b *ipfsBackendWithMetrics) WrapContextForRequest(ctx context.Context) context.Context {
+	if withCtxWrap, ok := b.backend.(WithContextHint); ok {
+		return withCtxWrap.WrapContextForRequest(ctx)
+	}
+	return ctx
+}
 
 func newHandlerWithMetrics(c *Config, backend IPFSBackend) *handler {
 	i := &handler{