mirror
/
viper
mirror of https://github.com/spf13/viper.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Add custom metrics for viper.Get calls

This commit is contained in:
Kyle Hagan 2020-04-10 16:11:02 -07:00
parent e34fb51dd7
commit f026d8b814
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
viper.go
View File

@ -213,7 +213,8 @@ type Viper struct {
// This will only be used if the configuration read is a properties file.
properties *properties.Properties
onConfigChange func(fsnotify.Event)
onConfigChange func(fsnotify.Event)
onGetCallMetric func(key string, value interface{})
}
// New returns an initialized Viper instance.
@ -537,6 +538,11 @@ func (v *Viper) AddSecureRemoteProvider(provider, endpoint, path, secretkeyring
return nil
}
func AddGetMetric(getCallMetric func(key string, val interface{})) { v.AddGetMetric(getCallMetric) }
func (v *Viper) AddGetMetric(getCallMetric func(key string, val interface{})) {
v.onGetCallMetric = getCallMetric
}
func (v *Viper) providerPathExists(p *defaultRemoteProvider) bool {
for _, y := range v.remoteProviders {
if reflect.DeepEqual(y, p) {
@ -726,6 +732,9 @@ func Get(key string) interface{} { return v.Get(key) }
func (v *Viper) Get(key string) interface{} {
lcaseKey := strings.ToLower(key)
val := v.find(lcaseKey, true)
if v.onGetCallMetric != nil {
v.onGetCallMetric(key, val)
}
if val == nil {
return nil
}