add metadata set method (#1232)

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2020-02-21 23:04:47 +03:00 committed by GitHub
parent 7e24c0c1cf
commit d1e25e7ead
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 22 additions and 0 deletions

View File

@ -22,6 +22,16 @@ func Copy(md Metadata) Metadata {
return cmd return cmd
} }
// Set add key with val to metadata
func Set(ctx context.Context, k, v string) context.Context {
md, ok := FromContext(ctx)
if !ok {
md = make(Metadata)
}
md[k] = v
return context.WithValue(ctx, metaKey{}, md)
}
// Get returns a single value from metadata in the context // Get returns a single value from metadata in the context
func Get(ctx context.Context, key string) (string, bool) { func Get(ctx context.Context, key string) (string, bool) {
md, ok := FromContext(ctx) md, ok := FromContext(ctx)

View File

@ -6,6 +6,18 @@ import (
"testing" "testing"
) )
func TestMetadataSet(t *testing.T) {
ctx := Set(context.TODO(), "Key", "val")
val, ok := Get(ctx, "Key")
if !ok {
t.Fatal("key Key not found")
}
if val != "val" {
t.Errorf("key Key with value val != %v", val)
}
}
func TestMetadataCopy(t *testing.T) { func TestMetadataCopy(t *testing.T) {
md := Metadata{ md := Metadata{
"Foo": "bar", "Foo": "bar",