forked from TrueCloudLab/frostfs-node
86 lines
1.7 KiB
Go
86 lines
1.7 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/TrueCloudLab/frostfs-node/cmd/frostfs-node/config/internal"
|
|
"github.com/TrueCloudLab/frostfs-node/pkg/util/config"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// Config represents a group of named values structured
|
|
// by tree type.
|
|
//
|
|
// Sub-trees are named configuration sub-sections,
|
|
// leaves are named configuration values.
|
|
// Names are of string type.
|
|
type Config struct {
|
|
v *viper.Viper
|
|
|
|
opts opts
|
|
|
|
defaultPath []string
|
|
path []string
|
|
}
|
|
|
|
const separator = "."
|
|
|
|
// Prm groups required parameters of the Config.
|
|
type Prm struct{}
|
|
|
|
// New creates a new Config instance.
|
|
//
|
|
// If file option is provided (WithConfigFile),
|
|
// configuration values are read from it.
|
|
// Otherwise, Config is a degenerate tree.
|
|
func New(_ Prm, opts ...Option) *Config {
|
|
v := viper.New()
|
|
|
|
v.SetEnvPrefix(internal.EnvPrefix)
|
|
v.AutomaticEnv()
|
|
v.SetEnvKeyReplacer(strings.NewReplacer(separator, internal.EnvSeparator))
|
|
|
|
o := defaultOpts()
|
|
for i := range opts {
|
|
opts[i](o)
|
|
}
|
|
|
|
if o.path != "" {
|
|
v.SetConfigFile(o.path)
|
|
|
|
err := v.ReadInConfig()
|
|
if err != nil {
|
|
panic(fmt.Errorf("failed to read config: %w", err))
|
|
}
|
|
}
|
|
|
|
if o.configDir != "" {
|
|
if err := config.ReadConfigDir(v, o.configDir); err != nil {
|
|
panic(fmt.Errorf("failed to read config dir: %w", err))
|
|
}
|
|
}
|
|
|
|
return &Config{
|
|
v: v,
|
|
opts: *o,
|
|
}
|
|
}
|
|
|
|
// Reload reads configuration path if it was provided to New.
|
|
func (x *Config) Reload() error {
|
|
if x.opts.path != "" {
|
|
err := x.v.ReadInConfig()
|
|
if err != nil {
|
|
return fmt.Errorf("rereading configuration file: %w", err)
|
|
}
|
|
}
|
|
|
|
if x.opts.configDir != "" {
|
|
if err := config.ReadConfigDir(x.v, x.opts.configDir); err != nil {
|
|
return fmt.Errorf("rereading configuration dir: %w", err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|