2016-03-15 21:38:16 +03:00
|
|
|
package hystrix
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/afex/hystrix-go/hystrix"
|
2020-01-31 01:26:39 +03:00
|
|
|
"github.com/micro/go-micro/v2/client"
|
2016-03-15 21:38:16 +03:00
|
|
|
|
2018-03-03 15:28:44 +03:00
|
|
|
"context"
|
2016-03-15 21:38:16 +03:00
|
|
|
)
|
|
|
|
|
|
|
|
type clientWrapper struct {
|
|
|
|
client.Client
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *clientWrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error {
|
2019-01-11 01:12:43 +03:00
|
|
|
return hystrix.Do(req.Service()+"."+req.Endpoint(), func() error {
|
2016-03-15 21:38:16 +03:00
|
|
|
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}
|
|
|
|
}
|
|
|
|
}
|