You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.3 KiB
67 lines
1.3 KiB
package metadata
|
|
|
|
import (
|
|
"strings"
|
|
)
|
|
|
|
// Metadata is our way of representing request headers internally.
|
|
// They're used at the RPC level and translate back and forth
|
|
// from Transport headers.
|
|
type Metadata map[string]string
|
|
|
|
// New creates an MD from a given key-values map.
|
|
func New(m map[string][]string) Metadata {
|
|
md := Metadata{}
|
|
for k, v := range m {
|
|
if k == "" {
|
|
continue
|
|
}
|
|
key := strings.ToLower(k)
|
|
if len(v) > 0 && v[0] != "" {
|
|
md[key] = v[0]
|
|
}
|
|
}
|
|
return md
|
|
}
|
|
|
|
// Get returns the value associated with the passed key.
|
|
func (m Metadata) Get(key string) string {
|
|
k := strings.ToLower(key)
|
|
return m[k]
|
|
}
|
|
|
|
// Set stores the key-value pair.
|
|
func (m Metadata) Set(key string, value string) {
|
|
if key == "" || value == "" {
|
|
return
|
|
}
|
|
k := strings.ToLower(key)
|
|
m[k] = value
|
|
}
|
|
|
|
// Keys lists the keys stored in this carrier.
|
|
func (m Metadata) Keys() []string {
|
|
keys := make([]string, 0, len(m))
|
|
for k := range m {
|
|
keys = append(keys, k)
|
|
}
|
|
return keys
|
|
}
|
|
|
|
// Pairs returns all metadata to key/value pairs.
|
|
func (m Metadata) Pairs() []string {
|
|
var kvs = make([]string, 0, len(m)*2)
|
|
for k, v := range m {
|
|
kvs = append(kvs, k, v)
|
|
}
|
|
return kvs
|
|
}
|
|
|
|
// Clone returns a deep copy of Metadata
|
|
func (m Metadata) Clone() Metadata {
|
|
md := Metadata{}
|
|
for k, v := range m {
|
|
md[k] = v
|
|
}
|
|
return md
|
|
}
|
|
|