github_mirrors/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2025-01-23 09:00:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

GetBreaker need double-check (#112)

Browse Source
This commit is contained in:
shenbaise9527 2020-09-30 16:50:02 +08:00 committed by GitHub
parent 0ce654968d
commit 9e9ce3bf48
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
core/breaker/breakers.go
View File

@ -41,10 +41,12 @@ func GetBreaker(name string) Breaker {
}
lock.Lock()
defer lock.Unlock()
b = NewBreaker()
breakers[name] = b
b, ok = breakers[name]
if !ok {
b = NewBreaker(WithName(name))
breakers[name] = b
}
lock.Unlock()
return b
}
@ -55,20 +57,5 @@ func NoBreakFor(name string) {
}
func do(name string, execute func(b Breaker) error) error {
lock.RLock()
b, ok := breakers[name]
lock.RUnlock()
if ok {
return execute(b)
}
lock.Lock()
b, ok = breakers[name]
if !ok {
b = NewBreaker(WithName(name))
breakers[name] = b
}
lock.Unlock()
return execute(b)
return execute(GetBreaker(name))
}