add sync.RWMutex for goroutine safe

This commit is contained in:
root 2019-10-31 14:45:35 +08:00
parent 40e41dd224
commit 69c351ace2
1 changed files with 11 additions and 1 deletions

View File

@ -59,7 +59,7 @@ func (e ConfigMarshalError) Error() string {
}
var v *Viper
var mutex sync.RWMutex
type RemoteResponse struct {
Value []byte
Error error
@ -499,7 +499,9 @@ func (v *Viper) searchMap(source map[string]interface{}, path []string) interfac
return source
}
mutex.RLock()
next, ok := source[path[0]]
mutex.RUnlock()
if ok {
// Fast path
if len(path) == 1 {
@ -542,7 +544,9 @@ func (v *Viper) searchMapWithPathPrefixes(source map[string]interface{}, path []
for i := len(path); i > 0; i-- {
prefixKey := strings.ToLower(strings.Join(path[0:i], v.keyDelim))
mutex.RLock()
next, ok := source[prefixKey]
mutex.RUnlock()
if ok {
// Fast path
if i == len(path) {
@ -1233,7 +1237,9 @@ func (v *Viper) Set(key string, value interface{}) {
deepestMap := deepSearch(v.override, path[0:len(path)-1])
// set innermost value
mutex.Lock()
deepestMap[lastKey] = value
mutex.Unlock()
}
// ReadInConfig will discover and load the configuration file from disk
@ -1442,7 +1448,9 @@ func (v *Viper) unmarshalReader(in io.Reader, c map[string]interface{}) error {
lastKey := strings.ToLower(path[len(path)-1])
deepestMap := deepSearch(c, path[0:len(path)-1])
// set innermost value
mutex.Lock()
deepestMap[lastKey] = value
mutex.Unlock()
}
}
@ -1805,7 +1813,9 @@ func (v *Viper) AllSettings() map[string]interface{} {
lastKey := strings.ToLower(path[len(path)-1])
deepestMap := deepSearch(m, path[0:len(path)-1])
// set innermost value
mutex.Lock()
deepestMap[lastKey] = value
mutex.Unlock()
}
return m
}