unistack-org/micro
4
0
Fork 3
Code Issues 19 Pull Requests 2 Actions 2 Packages Projects Releases 262 Wiki Activity

Make copy of metadata in setHeaders

Browse Source
This commit is contained in:
Asim 2016-11-02 17:21:53 +00:00
parent a36e042983
commit 7635bd9753
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
wrapper.go
View File

@ -13,15 +13,21 @@ type clientWrapper struct {
} }
func (c *clientWrapper) setHeaders(ctx context.Context) context.Context { func (c *clientWrapper) setHeaders(ctx context.Context) context.Context {
md, ok := metadata.FromContext(ctx) md := make(metadata.Metadata)
if !ok {
md = metadata.Metadata{} if mda, ok := metadata.FromContext(ctx); ok {
// make copy of metadata
for k, v := range mda {
md[k] = v
} }
}
for k, v := range c.headers { for k, v := range c.headers {
if _, ok := md[k]; !ok { if _, ok := md[k]; !ok {
md[k] = v md[k] = v
} }
} }
return metadata.NewContext(ctx, md) return metadata.NewContext(ctx, md)
} }

3
wrapper_test.go
View File

@ -38,8 +38,7 @@ func TestWrapper(t *testing.T) {
} }
ctx := metadata.NewContext(context.Background(), d.existing) ctx := metadata.NewContext(context.Background(), d.existing)
c.setHeaders(ctx) ctx = c.setHeaders(ctx)
md, _ := metadata.FromContext(ctx) md, _ := metadata.FromContext(ctx)
for k, v := range d.headers { for k, v := range d.headers {