149 lines
2.4 KiB
Go
149 lines
2.4 KiB
Go
package metadata
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/go-gost/core/metadata"
|
|
)
|
|
|
|
type mapMetadata map[string]any
|
|
|
|
func NewMetadata(m map[string]any) metadata.Metadata {
|
|
return mapMetadata(m)
|
|
}
|
|
|
|
func (m mapMetadata) IsExists(key string) bool {
|
|
_, ok := m[key]
|
|
return ok
|
|
}
|
|
|
|
func (m mapMetadata) Set(key string, value any) {
|
|
m[key] = value
|
|
}
|
|
|
|
func (m mapMetadata) Get(key string) any {
|
|
if m != nil {
|
|
return m[key]
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func GetBool(md metadata.Metadata, key string) (v bool) {
|
|
if md == nil || !md.IsExists(key) {
|
|
return
|
|
}
|
|
switch vv := md.Get(key).(type) {
|
|
case bool:
|
|
return vv
|
|
case int:
|
|
return vv != 0
|
|
case string:
|
|
v, _ = strconv.ParseBool(vv)
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
func GetInt(md metadata.Metadata, key string) (v int) {
|
|
if md == nil {
|
|
return
|
|
}
|
|
|
|
switch vv := md.Get(key).(type) {
|
|
case bool:
|
|
if vv {
|
|
v = 1
|
|
}
|
|
case int:
|
|
return vv
|
|
case string:
|
|
v, _ = strconv.Atoi(vv)
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
func GetFloat(md metadata.Metadata, key string) (v float64) {
|
|
if md == nil {
|
|
return
|
|
}
|
|
|
|
switch vv := md.Get(key).(type) {
|
|
case int:
|
|
return float64(vv)
|
|
case string:
|
|
v, _ = strconv.ParseFloat(vv, 64)
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
func GetDuration(md metadata.Metadata, key string) (v time.Duration) {
|
|
if md == nil {
|
|
return
|
|
}
|
|
switch vv := md.Get(key).(type) {
|
|
case int:
|
|
return time.Duration(vv) * time.Second
|
|
case string:
|
|
v, _ = time.ParseDuration(vv)
|
|
if v == 0 {
|
|
n, _ := strconv.Atoi(vv)
|
|
v = time.Duration(n) * time.Second
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func GetString(md metadata.Metadata, key string) (v string) {
|
|
if md != nil {
|
|
v, _ = md.Get(key).(string)
|
|
}
|
|
return
|
|
}
|
|
|
|
func GetStrings(md metadata.Metadata, key string) (ss []string) {
|
|
switch v := md.Get(key).(type) {
|
|
case []string:
|
|
ss = v
|
|
case []any:
|
|
for _, vv := range v {
|
|
if s, ok := vv.(string); ok {
|
|
ss = append(ss, s)
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func GetStringMap(md metadata.Metadata, key string) (m map[string]any) {
|
|
switch vv := md.Get(key).(type) {
|
|
case map[string]any:
|
|
return vv
|
|
case map[any]any:
|
|
m = make(map[string]any)
|
|
for k, v := range vv {
|
|
m[fmt.Sprintf("%v", k)] = v
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func GetStringMapString(md metadata.Metadata, key string) (m map[string]string) {
|
|
switch vv := md.Get(key).(type) {
|
|
case map[string]any:
|
|
m = make(map[string]string)
|
|
for k, v := range vv {
|
|
m[k] = fmt.Sprintf("%v", v)
|
|
}
|
|
case map[any]any:
|
|
m = make(map[string]string)
|
|
for k, v := range vv {
|
|
m[fmt.Sprintf("%v", k)] = fmt.Sprintf("%v", v)
|
|
}
|
|
}
|
|
return
|
|
}
|