go-pattern-examples/resiliency/01_circuit_breaker/circuit_counter.go
2020-06-05 12:21:52 +08:00

86 lines
1.7 KiB
Go

/*
* @Description: https://github.com/crazybber
* @Author: Edward
* @Date: 2020-05-22 12:41:54
* @Last Modified by: Edward
* @Last Modified time: 2020-05-22 17:22:35
*/
package circuit
import "time"
////////////////////////////////
/// 计数器 用以维护断路器内部的状态
/// 无论是对象式断路器还是函数式断路器
/// 都要用到计数器
////////////////////////////////
//State 断路器本身的状态的
//State of switch int
type State int
// state for breaker
const (
StateClosed State = iota //默认的闭合状态,可以正常执行业务
StateHalfOpen
StateOpen
StateUnknown
)
//OperationState of current 某一次操作的结果状态
type OperationState int
//ICounter interface
type ICounter interface {
Count(OperationState, bool)
LastActivity() time.Time
Reset()
Total() uint32
}
type counters struct {
Requests uint32 //连续的请求次数
lastActivity time.Time
TotalFailures uint32
TotalSuccesses uint32
ConsecutiveSuccesses uint32
ConsecutiveFailures uint32
}
func (c *counters) Total() uint32 {
return c.Requests
}
func (c *counters) LastActivity() time.Time {
return c.lastActivity
}
func (c *counters) Reset() {
ct := &counters{}
ct.lastActivity = c.lastActivity
c = ct
}
//Count the failure and success
func (c *counters) Count(statue OperationState, isConsecutive bool) {
switch statue {
case FailureState:
c.TotalFailures++
if isConsecutive || c.ConsecutiveFailures == 0 {
c.ConsecutiveFailures++
}
case SuccessState:
c.TotalSuccesses++
if isConsecutive || c.ConsecutiveSuccesses == 0 {
c.ConsecutiveSuccesses++
}
}
c.Requests++
c.lastActivity = time.Now() //更新活动时间
// c.lastOpResult = statue
//handle status change
}