diff --git a/go.mod b/go.mod index c305b54..efa0f94 100644 --- a/go.mod +++ b/go.mod @@ -2,4 +2,4 @@ module go.unistack.org/micro-wrapper-recovery/v3 go 1.16 -require go.unistack.org/micro/v3 v3.9.14 +require go.unistack.org/micro/v3 v3.9.13 diff --git a/go.sum b/go.sum index ca201b2..aeb4d06 100644 --- a/go.sum +++ b/go.sum @@ -74,8 +74,8 @@ github.com/xeipuuv/gojsonschema v1.2.0/go.mod h1:anYRn/JVcOK2ZgGU+IjEV4nwlhoK5sQ go.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI= go.unistack.org/micro-proto/v3 v3.3.1 h1:nQ0MtWvP2G3QrpOgawVOPhpZZYkq6umTGDqs8FxJYIo= go.unistack.org/micro-proto/v3 v3.3.1/go.mod h1:cwRyv8uInM2I7EbU7O8Fx2Ls3N90Uw9UCCcq4olOdfE= -go.unistack.org/micro/v3 v3.9.14 h1:KTH3mrAMALFLAzH0fLJpaVmKjEa0VlHMRBJBUCKD91g= -go.unistack.org/micro/v3 v3.9.14/go.mod h1:gI4RkJKHLPW7KV6h4+ZBOZD997MRvFRXMPQIHpozikI= +go.unistack.org/micro/v3 v3.9.13 h1:x1/H4rDgvz2JWXx6UvFiWDyRLxg2Kpp6V1kjzpLlyvo= +go.unistack.org/micro/v3 v3.9.13/go.mod h1:gI4RkJKHLPW7KV6h4+ZBOZD997MRvFRXMPQIHpozikI= golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA= diff --git a/wrapper.go b/wrapper.go index 2793239..9b37155 100644 --- a/wrapper.go +++ b/wrapper.go @@ -4,22 +4,60 @@ import ( "context" "fmt" + "go.unistack.org/micro/v3/errors" "go.unistack.org/micro/v3/server" ) -type wrapper struct { - serverHandlerFunc func(context.Context, server.Request, interface{}, error) error - serverSubscriberFunc func(context.Context, server.Message, error) error - /* - clientCallFunc func(context.Context, string, client.Request, interface{}, client.CallOptions, error) error - clientClient func(client.Client, error) error - */ +var ( + _ server.HandlerWrapper = NewServerHandlerWrapper() + _ server.SubscriberWrapper = NewServerSubscriberWrapper() +) + +func NewOptions(opts ...Option) Options { + options := Options{ + ServerHandlerFn: DefaultServerHandlerFn, + ServerSubscriberFn: DefaultServerSubscriberFn, + } + for _, o := range opts { + o(&options) + } + return options } -func NewServerHandlerWrapper(fn func(context.Context, server.Request, interface{}, error) error) server.HandlerWrapper { - handler := &wrapper{ - serverHandlerFunc: fn, +type Options struct { + ServerHandlerFn func(context.Context, server.Request, interface{}, error) error + ServerSubscriberFn func(context.Context, server.Message, error) error +} + +type Option func(*Options) + +func ServerHandlerFn(fn func(context.Context, server.Request, interface{}, error) error) Option { + return func(o *Options) { + o.ServerHandlerFn = fn } +} + +func ServerSubscriberFn(fn func(context.Context, server.Message, error) error) Option { + return func(o *Options) { + o.ServerSubscriberFn = fn + } +} + +var ( + DefaultServerHandlerFn = func(ctx context.Context, req server.Request, rsp interface{}, err error) error { + return errors.BadRequest("", "%v", err) + } + DefaultServerSubscriberFn = func(ctx context.Context, req server.Message, err error) error { + return errors.BadRequest("", "%v", err) + } +) + +type wrapper struct { + opts Options +} + +func NewServerHandlerWrapper(opts ...Option) server.HandlerWrapper { + handler := &wrapper{opts: NewOptions(opts...)} return handler.HandlerFunc } @@ -31,9 +69,9 @@ func (w *wrapper) HandlerFunc(fn server.HandlerFunc) server.HandlerFunc { case nil: return case error: - err = w.serverHandlerFunc(ctx, req, rsp, verr) + err = w.opts.ServerHandlerFn(ctx, req, rsp, verr) default: - err = w.serverHandlerFunc(ctx, req, rsp, fmt.Errorf("%v", r)) + err = w.opts.ServerHandlerFn(ctx, req, rsp, fmt.Errorf("%v", r)) } }() err = fn(ctx, req, rsp) @@ -41,10 +79,8 @@ func (w *wrapper) HandlerFunc(fn server.HandlerFunc) server.HandlerFunc { } } -func NewServerSubscriberWrapper(fn func(context.Context, server.Message, error) error) server.SubscriberWrapper { - handler := &wrapper{ - serverSubscriberFunc: fn, - } +func NewServerSubscriberWrapper(opts ...Option) server.SubscriberWrapper { + handler := &wrapper{opts: NewOptions(opts...)} return handler.SubscriberFunc } @@ -56,9 +92,9 @@ func (w *wrapper) SubscriberFunc(fn server.SubscriberFunc) server.SubscriberFunc case nil: return case error: - err = w.serverSubscriberFunc(ctx, msg, verr) + err = w.opts.ServerSubscriberFn(ctx, msg, verr) default: - err = w.serverSubscriberFunc(ctx, msg, fmt.Errorf("%v", r)) + err = w.opts.ServerSubscriberFn(ctx, msg, fmt.Errorf("%v", r)) } }() err = fn(ctx, msg)