29 lines
790 B
Go
29 lines
790 B
Go
|
package metadata
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"go-common/library/net/metadata"
|
||
|
|
)
|
||
|
|
|
||
|
|
// FromContext get metadata from context.
|
||
|
|
func FromContext(c context.Context) map[string]string {
|
||
|
|
return map[string]string{
|
||
|
|
metadata.Color: metadata.String(c, metadata.Color),
|
||
|
|
metadata.Caller: metadata.String(c, metadata.Caller),
|
||
|
|
metadata.Mirror: metadata.String(c, metadata.Mirror),
|
||
|
|
metadata.RemoteIP: metadata.String(c, metadata.RemoteIP),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewContext new metadata context.
|
||
|
|
func NewContext(c context.Context, meta map[string]string) context.Context {
|
||
|
|
md := metadata.MD{
|
||
|
|
metadata.Color: meta[metadata.Color],
|
||
|
|
metadata.Caller: meta[metadata.Caller],
|
||
|
|
metadata.Mirror: meta[metadata.Mirror],
|
||
|
|
metadata.RemoteIP: meta[metadata.RemoteIP],
|
||
|
|
}
|
||
|
|
return metadata.NewContext(c, md)
|
||
|
|
}
|