Pass outgoing gRPC metadata (#798)

Between contexts.
cirruslabs · Oct 2, 2024 · f20f699 · f20f699
1 parent 4eece3f
commit f20f699
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/internal/worker/task.go b/internal/worker/task.go
@@ -192,6 +192,10 @@ func (worker *Worker) runTask(
 		boundedCtx, cancel := context.WithTimeout(context.Background(), perCallTimeout)
 		defer cancel()
 
+		if md, ok := metadata.FromOutgoingContext(ctx); ok {
+			boundedCtx = metadata.NewOutgoingContext(boundedCtx, md)
+		}
+
 		localHub.WithScope(func(scope *sentry.Scope) {
 			scope.SetTags(cirrusSentryTags)
 
@@ -233,6 +237,10 @@ func (worker *Worker) runTask(
 	boundedCtx, cancel := context.WithTimeout(context.Background(), perCallTimeout)
 	defer cancel()
 
+	if md, ok := metadata.FromOutgoingContext(ctx); ok {
+		boundedCtx = metadata.NewOutgoingContext(boundedCtx, md)
+	}
+
 	if err = upstream.TaskStopped(boundedCtx, api.OldTaskIdentification(taskID, clientSecret)); err != nil {
 		worker.logger.Errorf("failed to notify the server about the stopped task %s: %v",
 			taskID, err)