unistack-org/micro
3
0
Fork 2
Code Issues 21 Pull Requests 2 Actions Packages Projects Releases 295 Wiki Activity

Rename to Merge

Browse Source
This commit is contained in:
Asim Aslam 2019-10-25 23:27:59 +01:00
parent 20c6c36bc4
commit 92b7d2db3b
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
metadata/metadata.go
View File

@ -12,6 +12,7 @@ type metaKey struct{}
// from Transport headers. // from Transport headers.
type Metadata map[string]string type Metadata map[string]string
// Copy makes a copy of the metadata
func Copy(md Metadata) Metadata { func Copy(md Metadata) Metadata {
cmd := make(Metadata) cmd := make(Metadata)
for k, v := range md { for k, v := range md {
@ -20,16 +21,19 @@ func Copy(md Metadata) Metadata {
return cmd return cmd
} }
// FromContext returns metadata from the given context
func FromContext(ctx context.Context) (Metadata, bool) { func FromContext(ctx context.Context) (Metadata, bool) {
md, ok := ctx.Value(metaKey{}).(Metadata) md, ok := ctx.Value(metaKey{}).(Metadata)
return md, ok return md, ok
} }
// NewContext creates a new context with the given metadata
func NewContext(ctx context.Context, md Metadata) context.Context { func NewContext(ctx context.Context, md Metadata) context.Context {
return context.WithValue(ctx, metaKey{}, md) return context.WithValue(ctx, metaKey{}, md)
} }
func AppendContext(ctx context.Context, patchMd Metadata, overwrite bool) context.Context { // MergeContext merges metadata to existing metadata, overwriting if specified
func MergeContext(ctx context.Context, patchMd Metadata, overwrite bool) context.Context {
md, _ := ctx.Value(metaKey{}).(Metadata) md, _ := ctx.Value(metaKey{}).(Metadata)
cmd := make(Metadata) cmd := make(Metadata)
for k, v := range md { for k, v := range md {