7f30a5e6e7
They should not be interpreted as 0, since that may overwrite defaults which are not 0. We simply need to do nothing. License: MIT Signed-off-by: Hector Sanjuan <code@hector.link>
117 lines
2.4 KiB
Go
117 lines
2.4 KiB
Go
package config
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
// Saver implements common functionality useful for ComponentConfigs
|
|
type Saver struct {
|
|
save chan struct{}
|
|
BaseDir string
|
|
}
|
|
|
|
// NotifySave signals the SaveCh() channel in a non-blocking fashion.
|
|
func (sv *Saver) NotifySave() {
|
|
if sv.save == nil {
|
|
sv.save = make(chan struct{}, 10)
|
|
}
|
|
|
|
// Non blocking, in case no one's listening
|
|
select {
|
|
case sv.save <- struct{}{}:
|
|
default:
|
|
logger.Warning("configuration save channel full")
|
|
}
|
|
}
|
|
|
|
// SaveCh returns a channel which is signaled when a component wants
|
|
// to persist its configuration
|
|
func (sv *Saver) SaveCh() <-chan struct{} {
|
|
if sv.save == nil {
|
|
sv.save = make(chan struct{})
|
|
}
|
|
return sv.save
|
|
}
|
|
|
|
// SetBaseDir is a setter for BaseDir and implements
|
|
// part of the ComponentConfig interface.
|
|
func (sv *Saver) SetBaseDir(dir string) {
|
|
sv.BaseDir = dir
|
|
}
|
|
|
|
// DefaultJSONMarshal produces pretty JSON with 2-space indentation
|
|
func DefaultJSONMarshal(v interface{}) ([]byte, error) {
|
|
bs, err := json.MarshalIndent(v, "", " ")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return bs, nil
|
|
}
|
|
|
|
// SetIfNotDefault sets dest to the value of src if src is not the default
|
|
// value of the type.
|
|
// dest must be a pointer.
|
|
func SetIfNotDefault(src interface{}, dest interface{}) {
|
|
switch src.(type) {
|
|
case time.Duration:
|
|
t := src.(time.Duration)
|
|
if t != 0 {
|
|
*dest.(*time.Duration) = t
|
|
}
|
|
case string:
|
|
str := src.(string)
|
|
if str != "" {
|
|
*dest.(*string) = str
|
|
}
|
|
case uint64:
|
|
n := src.(uint64)
|
|
if n != 0 {
|
|
*dest.(*uint64) = n
|
|
}
|
|
case int:
|
|
n := src.(int)
|
|
if n != 0 {
|
|
*dest.(*int) = n
|
|
}
|
|
case bool:
|
|
b := src.(bool)
|
|
if b {
|
|
*dest.(*bool) = b
|
|
}
|
|
}
|
|
}
|
|
|
|
// DurationOpt provides a datatype to use with ParseDurations
|
|
type DurationOpt struct {
|
|
// The duration we need to parse
|
|
Duration string
|
|
// Where to store the result
|
|
Dst *time.Duration
|
|
// A variable name associated to it for helpful errors.
|
|
Name string
|
|
}
|
|
|
|
// ParseDurations takes a time.Duration src and saves it to the given dst.
|
|
func ParseDurations(component string, args ...*DurationOpt) error {
|
|
for _, arg := range args {
|
|
if arg.Duration == "" {
|
|
// don't do anything. Let the destination field
|
|
// stay at its default.
|
|
continue
|
|
}
|
|
t, err := time.ParseDuration(arg.Duration)
|
|
if err != nil {
|
|
return fmt.Errorf(
|
|
"error parsing %s.%s: %s",
|
|
component,
|
|
arg.Name,
|
|
err,
|
|
)
|
|
}
|
|
*arg.Dst = t
|
|
}
|
|
return nil
|
|
}
|