diff options
Diffstat (limited to 'plugins/config/viper.go')
| -rw-r--r-- | plugins/config/viper.go | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/plugins/config/viper.go b/plugins/config/viper.go index 0362e79b..b276dbe2 100644 --- a/plugins/config/viper.go +++ b/plugins/config/viper.go @@ -17,17 +17,21 @@ type ViperProvider struct { //////// ENDURE ////////// func (v *ViperProvider) Init() error { v.viper = viper.New() + // read in environment variables that match v.viper.AutomaticEnv() if v.Prefix == "" { return errors.New("prefix should be set") } + v.viper.SetEnvPrefix(v.Prefix) if v.Path == "" { return errors.New("path should be set") } + v.viper.SetConfigFile(v.Path) v.viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) + return v.viper.ReadInConfig() } @@ -62,6 +66,11 @@ func (v *ViperProvider) Get(name string) interface{} { return v.viper.Get(name) } +// Has checks if config section exists. +func (v *ViperProvider) Has(name string) bool { + return v.viper.IsSet(name) +} + /////////// PRIVATE ////////////// func parseFlag(flag string) (string, string, error) { |