Add HTTP Tracecontext propagation to delivery client provider

google · May 14, 2020 · cafe654 · cafe654
1 parent 585f6d4
commit cafe654
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/cmd/broker/fanout/wire_gen.go b/cmd/broker/fanout/wire_gen.go
diff --git a/cmd/broker/retry/wire_gen.go b/cmd/broker/retry/wire_gen.go
diff --git a/pkg/broker/handler/pool/providers.go b/pkg/broker/handler/pool/providers.go
@@ -24,6 +24,8 @@ import (
 	cehttp "github.com/cloudevents/sdk-go/v2/protocol/http"
 	cepubsub "github.com/cloudevents/sdk-go/v2/protocol/pubsub"
 	"github.com/google/wire"
+	"go.opencensus.io/plugin/ochttp"
+	"go.opencensus.io/plugin/ochttp/propagation/tracecontext"
 )
 
 var (
@@ -33,6 +35,10 @@ var (
 		ceclient.WithTracePropagation(),
 	}
 
+	DefaultHTTPOpts = []cehttp.Option{
+		cehttp.WithRoundTripper(&ochttp.Transport{Propagation: &tracecontext.HTTPFormat{}}),
+	}
+
 	// ProviderSet provides the fanout and retry sync pools using the default client options. In
 	// order to inject either pool, ProjectID, []Option, and config.ReadOnlyTargets must be
 	// externally provided.
@@ -43,7 +49,7 @@ var (
 		NewDeliverClient,
 		NewPubsubClient,
 		NewRetryClient,
-		wire.Value([]cehttp.Option(nil)),
+		wire.Value(DefaultHTTPOpts),
 		wire.Value(DefaultCEClientOpts),
 	)
 )