fix: correct the context usage

snapp-incubator · Mar 3, 2024 · fa4dbfe · fa4dbfe
1 parent 4662956
commit fa4dbfe
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/internal/client/client.go b/internal/client/client.go
@@ -202,8 +202,7 @@ func (c *Client) Disconnect() {
 	c.Client.Disconnect(DisconnectTimeout)
 }
 
-func (c *Client) Connect() error {
-	ctx := otel.GetTextMapPropagator().Extract(context.Background(), nil)
+func (c *Client) Connect(ctx context.Context) error {
 	_, span := c.Tracer.Start(ctx, "client.on.connect")
 
 	defer span.End()

diff --git a/internal/cmd/main.go b/internal/cmd/main.go
@@ -25,7 +25,7 @@ func main(cfg config.Config, logger *zap.Logger, trace trace.Tracer) {
 
 		client := client.New(ctx, cfg.MQTT, logger.Named("mqtt"), trace, &c, true)
 
-		if err := client.Connect(); err != nil {
+		if err := client.Connect(ctx); err != nil {
 			span.RecordError(err)
 			span.SetStatus(codes.Error, err.Error())
 
@@ -39,7 +39,7 @@ func main(cfg config.Config, logger *zap.Logger, trace trace.Tracer) {
 
 		client := client.New(ctx, cfg.MQTT, logger.Named("mqtt"), trace, &c, false)
 
-		if err := client.Connect(); err != nil {
+		if err := client.Connect(ctx); err != nil {
 			span.RecordError(err)
 			span.SetStatus(codes.Error, err.Error())