stats: create new MD for outgoing header if no MD is available (#3357)

grpc · Feb 4, 2020 · 3ae60eb · 3ae60eb
1 parent e0e0cbc
commit 3ae60eb
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/internal/transport/http2_client.go b/internal/transport/http2_client.go
@@ -680,8 +680,12 @@ func (t *http2Client) NewStream(ctx context.Context, callHdr *CallHdr) (_ *Strea
 		}
 	}
 	if t.statsHandler != nil {
-		header, _ := metadata.FromOutgoingContext(ctx)
-		header.Set("user-agent", t.userAgent)
+		header, ok := metadata.FromOutgoingContext(ctx)
+		if ok {
+			header.Set("user-agent", t.userAgent)
+		} else {
+			header = metadata.Pairs("user-agent", t.userAgent)
+		}
 		outHeader := &stats.OutHeader{
 			Client:      true,
 			FullMethod:  callHdr.Method,