gobreaker

type CircuitBreaker ¶

type CircuitBreaker[T any] struct {
	// contains filtered or unexported fields
}

CircuitBreaker is a state machine to prevent sending requests that are likely to fail.

func NewCircuitBreaker[T any](st Settings) *CircuitBreaker[T]

func (cb *CircuitBreaker[T]) Counts() Counts

func (cb *CircuitBreaker[T]) Execute(req func() (T, error)) (T, error)

func (cb *CircuitBreaker[T]) Name() string

func (cb *CircuitBreaker[T]) State() State

type Counts ¶

type Counts struct {
	Requests             uint32
	TotalSuccesses       uint32
	TotalFailures        uint32
	ConsecutiveSuccesses uint32
	ConsecutiveFailures  uint32
}

Counts holds the numbers of requests and their successes/failures. CircuitBreaker clears the internal Counts either on the change of the state or at the closed-state intervals. Counts ignores the results of the requests sent before clearing.

type DistributedCircuitBreaker ¶

type DistributedCircuitBreaker[T any] struct {
	*CircuitBreaker[T]
	// contains filtered or unexported fields
}

DistributedCircuitBreaker extends CircuitBreaker with SharedDataStore.

func NewDistributedCircuitBreaker[T any](store SharedDataStore, settings Settings) (dcb *DistributedCircuitBreaker[T], err error)

func (dcb *DistributedCircuitBreaker[T]) Execute(req func() (T, error)) (t T, err error)

func (dcb *DistributedCircuitBreaker[T]) State() (state State, err error)

type RateLimitedCircuitBreaker ¶

type RateLimitedCircuitBreaker[T any] struct {
	// contains filtered or unexported fields
}

RateLimitedCircuitBreaker is a state machine to prevent sending requests that are likely to fail. It differs from a regular CircuitBreaker in that it has a half-open state that: 1. Allows a rate-limited number of requests to pass through, rejecting requests when the limit is reached 2. Resets the state to closed after a specified timeout 3. Uses the same tripping logic as the Closed state.

func NewRateLimitedCircuitBreaker[T any](st RateLimitedCircuitBreakerSettings) *RateLimitedCircuitBreaker[T]

func (cb *RateLimitedCircuitBreaker[T]) Counts() Counts

func (cb *RateLimitedCircuitBreaker[T]) Execute(req func() (T, error)) (T, error)

func (cb *RateLimitedCircuitBreaker[T]) Name() string

func (cb *RateLimitedCircuitBreaker[T]) State() State

func (cb *RateLimitedCircuitBreaker[T]) String() string

type RateLimitedCircuitBreakerSettings ¶

type RateLimitedCircuitBreakerSettings struct {
	Settings

	HalfOpenTimeout time.Duration
}

RateLimitedCircuitBreakerSettings configures RateLimitedCircuitBreaker:

Settings is the base configuration for the RateLimitedCircuitBreaker.

HalfOpenTimeout is the period of the halg-open state, after which the state of the RateLimitedCircuitBreaker becomes closed. If HalfOpenTimeout is less than or equal to 0, the timeout value of the RateLimitedCircuitBreaker is set to 60 seconds.

type RedisStore ¶

type RedisStore struct {
	// contains filtered or unexported fields
}

func NewRedisStore(addr string) *RedisStore

func (rs *RedisStore) Close()

func (rs *RedisStore) GetData(name string) ([]byte, error)

func (rs *RedisStore) Lock(name string) error

func (rs *RedisStore) SetData(name string, data []byte) error

func (rs *RedisStore) Unlock(name string) error

type Settings ¶

type Settings struct {
	Name          string
	MaxRequests   uint32
	Interval      time.Duration
	Timeout       time.Duration
	ReadyToTrip   func(counts Counts) bool
	OnStateChange func(name string, from State, to State)
	IsSuccessful  func(err error) bool
}

Settings configures CircuitBreaker:

Name is the name of the CircuitBreaker.

MaxRequests is the maximum number of requests allowed to pass through when the CircuitBreaker is half-open. If MaxRequests is 0, the CircuitBreaker allows only 1 request.

Interval is the cyclic period of the closed state for the CircuitBreaker to clear the internal Counts. If Interval is less than or equal to 0, the CircuitBreaker doesn't clear internal Counts during the closed state.

Timeout is the period of the open state, after which the state of the CircuitBreaker becomes half-open. If Timeout is less than or equal to 0, the timeout value of the CircuitBreaker is set to 60 seconds.

ReadyToTrip is called with a copy of Counts whenever a request fails in the closed state. If ReadyToTrip returns true, the CircuitBreaker will be placed into the open state. If ReadyToTrip is nil, default ReadyToTrip is used. Default ReadyToTrip returns true when the number of consecutive failures is more than 5.

OnStateChange is called whenever the state of the CircuitBreaker changes.

IsSuccessful is called with the error returned from a request. If IsSuccessful returns true, the error is counted as a success. Otherwise the error is counted as a failure. If IsSuccessful is nil, default IsSuccessful is used, which returns false for all non-nil errors.

type SharedDataStore ¶

type SharedDataStore interface {
	Lock(name string) error
	Unlock(name string) error
	GetData(name string) ([]byte, error)
	SetData(name string, data []byte) error
}

SharedDataStore stores the shared state of DistributedCircuitBreaker.

type SharedState ¶

type SharedState struct {
	State      State     `json:"state"`
	Generation uint64    `json:"generation"`
	Counts     Counts    `json:"counts"`
	Expiry     time.Time `json:"expiry"`
}

SharedState represents the shared state of DistributedCircuitBreaker.

type State ¶

type State int

State is a type that represents a state of CircuitBreaker.

const (
	StateClosed State = iota
	StateHalfOpen
	StateOpen
)

func (s State) String() string

type TwoStepCircuitBreaker ¶

type TwoStepCircuitBreaker[T any] struct {
	// contains filtered or unexported fields
}

TwoStepCircuitBreaker is like CircuitBreaker but instead of surrounding a function with the breaker functionality, it only checks whether a request can proceed and expects the caller to report the outcome in a separate step using a callback.

func NewTwoStepCircuitBreaker[T any](st Settings) *TwoStepCircuitBreaker[T]

func (tscb *TwoStepCircuitBreaker[T]) Allow() (done func(success bool), err error)

func (tscb *TwoStepCircuitBreaker[T]) Counts() Counts

func (tscb *TwoStepCircuitBreaker[T]) Name() string

func (tscb *TwoStepCircuitBreaker[T]) State() State