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

2020-02-04 08:07:17 -08:00
parent e0e0cbc795
commit 3ae60ebf74
1 changed files with 6 additions and 2 deletions
--- 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, 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,