mirror/go-pattern-examples
1
0
Fork 0
mirror of https://github.com/crazybber/go-pattern-examples.git synced 2025-02-16 17:23:15 +03:00
Code Issues Packages Projects Releases Wiki Activity

add state flow check

Browse Source
This commit is contained in:
Edward 2020-06-03 23:55:51 +08:00
parent 57c945fd94
commit c10a20aa5d
2 changed files with 25 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
resiliency/06_circuit_breaker/breaker_options.go
View File

@ -3,7 +3,7 @@
* @Author: Edward * @Author: Edward
* @Date: 2020-06-02 23:57:40 * @Date: 2020-06-02 23:57:40
* @Last Modified by: Edward * @Last Modified by: Edward
* @Last Modified time: 2020-06-02 23:57:40 * @Last Modified time: 2020-06-03 23:50:17
*/ */
package circuit package circuit
@ -30,6 +30,7 @@ type Options struct {
MaxRequests uint32 MaxRequests uint32
WhenToBreak BreakConditionWatcher //是否应该断开电路(打开电路开关) WhenToBreak BreakConditionWatcher //是否应该断开电路(打开电路开关)
OnStateChanged StateChangedEventHandler OnStateChanged StateChangedEventHandler
ShoulderToOpen uint32
Ctx context.Context Ctx context.Context
} }

31
resiliency/06_circuit_breaker/circuit_breaker_adv.go
View File

@ -3,7 +3,7 @@
* @Author: Edward * @Author: Edward
* @Date: 2020-05-10 22:00:58 * @Date: 2020-05-10 22:00:58
* @Last Modified by: Edward * @Last Modified by: Edward
* @Last Modified time: 2020-05-22 17:25:56 * @Last Modified time: 2020-06-03 23:54:29
*/ */
package circuit package circuit
@ -85,18 +85,14 @@ func (rb *RequestBreaker) Do(work func(ctx context.Context) (interface{}, error)
//before //before
fmt.Println("before do : request:", rb.cnter.Total()) fmt.Println("before do : request:", rb.cnter.Total())
rb.mutex.Lock() rb.mutex.Lock()
switch rb.state {
case StateOpen:
return nil, ErrTooManyRequests
case StateHalfOpen:
//如果是断开状态，并且超时了，转到半开状态，记录 //如果是断开状态，并且超时了，转到半开状态，记录
if rb.options.Expiry.Before(time.Now()) { if rb.state == StateOpen && rb.options.Expiry.Before(time.Now()) {
rb.state = StateHalfOpen
preState = rb.state preState = rb.state
rb.state = StateHalfOpen
rb.cnter.Reset() rb.cnter.Reset()
} else {
} return nil, ErrTooManyRequests
} }
rb.mutex.Unlock() rb.mutex.Unlock()
@ -112,17 +108,28 @@ func (rb *RequestBreaker) Do(work func(ctx context.Context) (interface{}, error)
//如果是在半开状态下的失败，立即打开开关 //如果是在半开状态下的失败，立即打开开关
if rb.state == StateHalfOpen { if rb.state == StateHalfOpen {
rb.state = StateOpen //转为打开 rb.state = StateOpen //转为打开
rb.options.OnStateChanged(rb.options.Name, StateHalfOpen, StateOpen) //半开到打开
} else if rb.state == StateClosed { } else if rb.state == StateClosed {
if rb.options.WhenToBreak(rb.cnter) { if rb.options.WhenToBreak(rb.cnter) {
rb.state = StateOpen //打开开关 rb.state = StateOpen //打开开关
rb.options.OnStateChanged(rb.options.Name, StateClosed, StateOpen) //关闭到打开
rb.cnter.Reset() rb.cnter.Reset()
} }
} }
} else { } else {
//成功了.
rb.cnter.Count(SuccessState) rb.cnter.Count(SuccessState)
if preState == StateOpen && rb.state == StateHalfOpen { if rb.state == StateHalfOpen {
rb.options.OnStateChanged(rb.options.Name, StateOpen, StateHalfOpen) if preState == StateOpen {
preState = StateHalfOpen
rb.options.OnStateChanged(rb.options.Name, StateOpen, StateHalfOpen) //打开到半开
}
if rb.cnter.ConsecutiveSuccesses >= rb.options.ShoulderToOpen {
rb.state = StateClosed
rb.options.OnStateChanged(rb.options.Name, StateHalfOpen, StateClosed) //半开到关闭
}
} }
} }