opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Menghan Li
2020-02-04 08:07:17 -08:00
committed by GitHub
parent e0e0cbc795
commit 3ae60ebf74
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/transport/http2_client.go
View File

@ -680,8 +680,12 @@ func (t *http2Client) NewStream(ctx context.Context, callHdr *CallHdr) (_ *Strea
} }
} }
if t.statsHandler != nil { if t.statsHandler != nil {
header, _ := metadata.FromOutgoingContext(ctx) header, ok := metadata.FromOutgoingContext(ctx)
header.Set("user-agent", t.userAgent) if ok {
header.Set("user-agent", t.userAgent)
} else {
header = metadata.Pairs("user-agent", t.userAgent)
}
outHeader := &stats.OutHeader{ outHeader := &stats.OutHeader{
Client: true, Client: true,
FullMethod: callHdr.Method, FullMethod: callHdr.Method,