package config import ( "encoding/json" "fmt" "reflect" "strconv" "sync/atomic" "time" "google.golang.org/protobuf/encoding/protojson" "google.golang.org/protobuf/proto" ) var ( _ Value = (*atomicValue)(nil) _ Value = (*errValue)(nil) ) // Value is config value interface. type Value interface { Bool() (bool, error) Int() (int64, error) Float() (float64, error) String() (string, error) Duration() (time.Duration, error) Scan(interface{}) error Load() interface{} Store(interface{}) } type atomicValue struct { atomic.Value } func (v *atomicValue) Bool() (bool, error) { switch val := v.Load().(type) { case bool: return val, nil case int, int32, int64, float64, string: return strconv.ParseBool(fmt.Sprint(val)) } return false, fmt.Errorf("type assert to %v failed", reflect.TypeOf(v.Load())) } func (v *atomicValue) Int() (int64, error) { switch val := v.Load().(type) { case int: return int64(val), nil case int32: return int64(val), nil case int64: return int64(val), nil case float64: return int64(val), nil case string: return strconv.ParseInt(val, 10, 64) } return 0, fmt.Errorf("type assert to %v failed", reflect.TypeOf(v.Load())) } func (v *atomicValue) Float() (float64, error) { switch val := v.Load().(type) { case float64: return float64(val), nil case int: return float64(val), nil case int32: return float64(val), nil case int64: return float64(val), nil case string: return strconv.ParseFloat(val, 64) } return 0.0, fmt.Errorf("type assert to %v failed", reflect.TypeOf(v.Load())) } func (v *atomicValue) String() (string, error) { switch val := v.Load().(type) { case string: return val, nil case bool, int, int32, int64, float64: return fmt.Sprint(val), nil default: if s, ok := val.(fmt.Stringer); ok { return s.String(), nil } } return "", fmt.Errorf("type assert to %v failed", reflect.TypeOf(v.Load())) } func (v *atomicValue) Duration() (time.Duration, error) { val, err := v.Int() if err != nil { return 0, err } return time.Duration(val), nil } func (v *atomicValue) Scan(obj interface{}) error { data, err := json.Marshal(v.Load()) if err != nil { return err } if pb, ok := obj.(proto.Message); ok { return protojson.Unmarshal(data, pb) } return json.Unmarshal(data, obj) } type errValue struct { err error } func (v errValue) Bool() (bool, error) { return false, v.err } func (v errValue) Int() (int64, error) { return 0, v.err } func (v errValue) Float() (float64, error) { return 0.0, v.err } func (v errValue) Duration() (time.Duration, error) { return 0, v.err } func (v errValue) String() (string, error) { return "", v.err } func (v errValue) Scan(interface{}) error { return v.err } func (v errValue) Load() interface{} { return nil } func (v errValue) Store(interface{}) {}