feat: inject context in header

mgjules · Apr 11, 2022 · 040e79f · 040e79f
1 parent 8060c71
commit 040e79f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/http/client.go b/http/client.go
@@ -4,6 +4,8 @@ import (
 	"github.com/JulesMike/spoty/config"
 	"github.com/JulesMike/spoty/json"
 	"github.com/go-resty/resty/v2"
+	"go.opentelemetry.io/otel"
+	"go.opentelemetry.io/otel/propagation"
 )
 
 // Client is a simple wrapper around resty.Client.
@@ -20,6 +22,11 @@ func NewClient(cfg *config.Config) *Client {
 		client.EnableTrace()
 		client.SetDebug(true)
 	}
+	client.OnBeforeRequest(func(_ *resty.Client, r *resty.Request) error {
+		otel.GetTextMapPropagator().Inject(r.Context(), propagation.HeaderCarrier(r.Header))
+
+		return nil
+	})
 
 	return &Client{client}
 }