unistack-org/micro-wrapper-validator
4
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases 4 Wiki Activity

Merge pull request #51 from unistack-org/validate_rsp

Browse Source 
enable rsp validation
This commit is contained in:
Василий Толстов 2022-12-03 02:28:17 +03:00 committed by GitHub
commit d1096636f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 37 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
validator.go
View File

@ -9,11 +9,17 @@ import (
) )
var ( var (
DefaultClientErrorFunc = func(req client.Request, err error) error { DefaultClientErrorFunc = func(req client.Request, rsp interface{}, err error) error {
if rsp != nil {
return errors.BadGateway(req.Service(), "%v", err)
}
return errors.BadRequest(req.Service(), "%v", err) return errors.BadRequest(req.Service(), "%v", err)
} }
DefaultServerErrorFunc = func(req server.Request, err error) error { DefaultServerErrorFunc = func(req server.Request, rsp interface{}, err error) error {
if rsp != nil {
return errors.BadGateway(req.Service(), "%v", err)
}
return errors.BadRequest(req.Service(), "%v", err) return errors.BadRequest(req.Service(), "%v", err)
} }
@ -27,8 +33,8 @@ var (
) )
type ( type (
ClientErrorFunc func(client.Request, error) error ClientErrorFunc func(client.Request, interface{}, error) error
ServerErrorFunc func(server.Request, error) error ServerErrorFunc func(server.Request, interface{}, error) error
PublishErrorFunc func(client.Message, error) error PublishErrorFunc func(client.Message, error) error
SubscribeErrorFunc func(server.Message, error) error SubscribeErrorFunc func(server.Message, error) error
) )
@ -44,7 +50,7 @@ type Options struct {
// Option func signature // Option func signature
type Option func(*Options) type Option func(*Options)
func ClientErrorFn(fn ClientErrorFunc) Option { func ClientReqErrorFn(fn ClientErrorFunc) Option {
return func(o *Options) { return func(o *Options) {
o.ClientErrorFn = fn o.ClientErrorFn = fn
} }
@ -105,11 +111,17 @@ func NewClientCallWrapper(opts ...Option) client.CallWrapper {
return func(fn client.CallFunc) client.CallFunc { return func(fn client.CallFunc) client.CallFunc {
return func(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) error { return func(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) error {
if v, ok := req.Body().(validator); ok { if v, ok := req.Body().(validator); ok {
if err := v.Validate(); err != nil { if verr := v.Validate(); verr != nil {
return options.ClientErrorFn(req, err) return options.ClientErrorFn(req, nil, verr)
} }
} }
return fn(ctx, addr, req, rsp, opts) err := fn(ctx, addr, req, rsp, opts)
if v, ok := rsp.(validator); ok {
if verr := v.Validate(); verr != nil {
return options.ClientErrorFn(req, rsp, verr)
}
}
return err
} }
} }
} }
@ -117,16 +129,22 @@ func NewClientCallWrapper(opts ...Option) client.CallWrapper {
func (w *wrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { func (w *wrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error {
if v, ok := req.Body().(validator); ok { if v, ok := req.Body().(validator); ok {
if err := v.Validate(); err != nil { if err := v.Validate(); err != nil {
return w.opts.ClientErrorFn(req, err) return w.opts.ClientErrorFn(req, nil, err)
} }
} }
return w.Client.Call(ctx, req, rsp, opts...) err := w.Client.Call(ctx, req, rsp, opts...)
if v, ok := rsp.(validator); ok {
if verr := v.Validate(); verr != nil {
return w.opts.ClientErrorFn(req, rsp, verr)
}
}
return err
} }
func (w *wrapper) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { func (w *wrapper) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) {
if v, ok := req.Body().(validator); ok { if v, ok := req.Body().(validator); ok {
if err := v.Validate(); err != nil { if err := v.Validate(); err != nil {
return nil, w.opts.ClientErrorFn(req, err) return nil, w.opts.ClientErrorFn(req, nil, err)
} }
} }
return w.Client.Stream(ctx, req, opts...) return w.Client.Stream(ctx, req, opts...)
@ -147,10 +165,16 @@ func NewServerHandlerWrapper(opts ...Option) server.HandlerWrapper {
return func(ctx context.Context, req server.Request, rsp interface{}) error { return func(ctx context.Context, req server.Request, rsp interface{}) error {
if v, ok := req.Body().(validator); ok { if v, ok := req.Body().(validator); ok {
if err := v.Validate(); err != nil { if err := v.Validate(); err != nil {
return options.ClientErrorFn(req, err) return options.ClientErrorFn(req, nil, err)
} }
} }
return fn(ctx, req, rsp) err := fn(ctx, req, rsp)
if v, ok := rsp.(validator); ok {
if verr := v.Validate(); verr != nil {
return options.ClientErrorFn(req, rsp, err)
}
}
return err
} }
} }
} }