add:viper: allow to disable internal log messages

This commit is contained in:
Quentin Burgess 2023-01-03 14:58:15 +01:00
parent 9da88d94fa
commit 27865b0daf

View file

@ -205,6 +205,7 @@ type Viper struct {
automaticEnvApplied bool
envKeyReplacer StringReplacer
allowEmptyEnv bool
logMessage bool
config map[string]interface{}
override map[string]interface{}
@ -240,6 +241,7 @@ func New() *Viper {
v.aliases = make(map[string]string)
v.typeByDefValue = false
v.logger = jwwLogger{}
v.logMessage = true
v.resetEncoding()
@ -268,6 +270,14 @@ func KeyDelimiter(d string) Option {
})
}
// DisableMessageLog block any kind of messages to be logged.
// By default, all messages are logged.
func DisableMessageLog() Option {
return optionFunc(func(v *Viper) {
v.logMessage = false
})
}
// StringReplacer applies a set of replacements to a string.
type StringReplacer interface {
// Replace returns a copy of s with all replacements performed.
@ -433,14 +443,16 @@ func (v *Viper) WatchConfig() {
initWG.Add(1)
go func() {
watcher, err := newWatcher()
if err != nil {
if err != nil && v.logMessage {
log.Fatal(err)
}
defer watcher.Close()
// we have to watch the entire directory to pick up renames/atomic saves in a cross-platform way
filename, err := v.getConfigFile()
if err != nil {
log.Printf("error: %v\n", err)
if v.logMessage {
log.Printf("error: %v\n", err)
}
initWG.Done()
return
}
@ -468,7 +480,7 @@ func (v *Viper) WatchConfig() {
(currentConfigFile != "" && currentConfigFile != realConfigFile) {
realConfigFile = currentConfigFile
err := v.ReadInConfig()
if err != nil {
if err != nil && v.logMessage {
log.Printf("error reading config file: %v\n", err)
}
if v.onConfigChange != nil {
@ -480,7 +492,7 @@ func (v *Viper) WatchConfig() {
}
case err, ok := <-watcher.Errors:
if ok { // 'Errors' channel is not closed
if ok && v.logMessage { // 'Errors' channel is not closed
log.Printf("watcher error: %v\n", err)
}
eventsWG.Done()