gsh-digital-services/seaweedFS
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

better locks

Browse Source
This commit is contained in:
Chris Lu
2021-01-13 13:20:33 -08:00
parent 819a85f59a
commit e2c7e3fe6d
1 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
weed/util/config.go
View File

@@ -52,6 +52,10 @@ type ViperProxy struct {
sync.Mutex sync.Mutex
} }
var (
vp = &ViperProxy{}
)
func (vp *ViperProxy) SetDefault(key string, value interface{}) { func (vp *ViperProxy) SetDefault(key string, value interface{}) {
vp.Lock() vp.Lock()
defer vp.Unlock() defer vp.Unlock()
@@ -83,10 +87,15 @@ func (vp *ViperProxy) GetStringSlice(key string) []string {
} }
func GetViper() *ViperProxy { func GetViper() *ViperProxy {
v := &ViperProxy{} vp.Lock()
v.Viper = viper.GetViper() defer vp.Unlock()
v.AutomaticEnv()
v.SetEnvPrefix("weed") if vp.Viper == nil {
v.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) vp.Viper = viper.GetViper()
return v vp.AutomaticEnv()
vp.SetEnvPrefix("weed")
vp.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
}
return vp
} }