Rename to wrapper

This commit is contained in:
Asim 2016-03-15 18:38:16 +00:00 committed by Vasiliy Tolstov
commit e9508fdeea
2 changed files with 69 additions and 0 deletions

25
hystrix.go Normal file
View File

@ -0,0 +1,25 @@
package hystrix
import (
"github.com/afex/hystrix-go/hystrix"
"github.com/micro/go-micro/client"
"golang.org/x/net/context"
)
type clientWrapper struct {
client.Client
}
func (c *clientWrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error {
return hystrix.Do(req.Service()+"."+req.Method(), func() error {
return c.Client.Call(ctx, req, rsp, opts...)
}, nil)
}
// NewClientWrapper returns a hystrix client Wrapper.
func NewClientWrapper() client.Wrapper {
return func(c client.Client) client.Client {
return &clientWrapper{c}
}
}

44
hystrix_test.go Normal file
View File

@ -0,0 +1,44 @@
package hystrix
import (
"testing"
"github.com/micro/go-micro/client"
"github.com/micro/go-micro/registry/mock"
"github.com/micro/go-micro/selector"
"golang.org/x/net/context"
)
func TestBreaker(t *testing.T) {
// setup
r := mock.NewRegistry()
s := selector.NewSelector(selector.Registry(r))
c := client.NewClient(
// set the selector
client.Selector(s),
// add the breaker wrapper
client.Wrap(NewClientWrapper()),
)
req := c.NewJsonRequest("test.service", "Test.Method", map[string]string{
"foo": "bar",
})
var rsp map[string]interface{}
// Force to point of trip
for i := 0; i < 25; i++ {
c.Call(context.TODO(), req, rsp)
}
err := c.Call(context.TODO(), req, rsp)
if err == nil {
t.Error("Expecting tripped breaker, got nil error")
}
if err.Error() != "hystrix: circuit open" {
t.Error("Expecting tripped breaker, got %v", err)
}
}