39 lines
944 B
Go
39 lines
944 B
Go
|
package circuitbreaker
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/streadway/handy/breaker"
|
||
|
|
|
||
|
|
"github.com/go-kit/kit/endpoint"
|
||
|
|
)
|
||
|
|
|
||
|
|
// HandyBreaker returns an endpoint.Middleware that implements the circuit
|
||
|
|
// breaker pattern using the streadway/handy/breaker package. Only errors
|
||
|
|
// returned by the wrapped endpoint count against the circuit breaker's error
|
||
|
|
// count.
|
||
|
|
//
|
||
|
|
// See http://godoc.org/github.com/streadway/handy/breaker for more
|
||
|
|
// information.
|
||
|
|
func HandyBreaker(cb breaker.Breaker) endpoint.Middleware {
|
||
|
|
return func(next endpoint.Endpoint) endpoint.Endpoint {
|
||
|
|
return func(ctx context.Context, request interface{}) (response interface{}, err error) {
|
||
|
|
if !cb.Allow() {
|
||
|
|
return nil, breaker.ErrCircuitOpen
|
||
|
|
}
|
||
|
|
|
||
|
|
defer func(begin time.Time) {
|
||
|
|
if err == nil {
|
||
|
|
cb.Success(time.Since(begin))
|
||
|
|
} else {
|
||
|
|
cb.Failure(time.Since(begin))
|
||
|
|
}
|
||
|
|
}(time.Now())
|
||
|
|
|
||
|
|
response, err = next(ctx, request)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|