63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
package config
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
|
|
"github.com/pkg/errors"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
// Config definition
|
|
type Config struct {
|
|
Logger LoggerConfig `yaml:"logger"`
|
|
Server ServerConfig `yaml:"server"`
|
|
Agent AgentConfig `yaml:"agent"`
|
|
}
|
|
|
|
// NewFromFile retrieves the configuration from the given file
|
|
func NewFromFile(path string) (*Config, error) {
|
|
config := NewDefault()
|
|
|
|
data, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "could not read file '%s'", path)
|
|
}
|
|
|
|
if err := yaml.Unmarshal(data, config); err != nil {
|
|
return nil, errors.Wrapf(err, "could not unmarshal configuration")
|
|
}
|
|
|
|
return config, nil
|
|
}
|
|
|
|
// NewDumpDefault dump the new default configuration
|
|
func NewDumpDefault() *Config {
|
|
config := NewDefault()
|
|
|
|
return config
|
|
}
|
|
|
|
// NewDefault return new default configuration
|
|
func NewDefault() *Config {
|
|
return &Config{
|
|
Logger: NewDefaultLoggerConfig(),
|
|
Agent: NewDefaultAgentConfig(),
|
|
Server: NewDefaultServerConfig(),
|
|
}
|
|
}
|
|
|
|
// Dump the given configuration in the given writer
|
|
func Dump(config *Config, w io.Writer) error {
|
|
data, err := yaml.Marshal(config)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not dump config")
|
|
}
|
|
|
|
if _, err := w.Write(data); err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
return nil
|
|
}
|