Codebase list golang-github-go-kit-kit / 93fed9e2-784b-413e-a591-aaae2048677d/main circuitbreaker / hystrix.go
93fed9e2-784b-413e-a591-aaae2048677d/main

Tree @93fed9e2-784b-413e-a591-aaae2048677d/main (Download .tar.gz)

hystrix.go @93fed9e2-784b-413e-a591-aaae2048677d/mainraw · history · blame

package circuitbreaker

import (
	"context"

	"github.com/afex/hystrix-go/hystrix"

	"github.com/go-kit/kit/endpoint"
)

// Hystrix returns an endpoint.Middleware that implements the circuit
// breaker pattern using the afex/hystrix-go package.
//
// When using this circuit breaker, please configure your commands separately.
//
// See https://godoc.org/github.com/afex/hystrix-go/hystrix for more
// information.
func Hystrix(commandName string) endpoint.Middleware {
	return func(next endpoint.Endpoint) endpoint.Endpoint {
		return func(ctx context.Context, request interface{}) (response interface{}, err error) {
			var resp interface{}
			if err := hystrix.Do(commandName, func() (err error) {
				resp, err = next(ctx, request)
				return err
			}, nil); err != nil {
				return nil, err
			}
			return resp, nil
		}
	}
}