1
0
mirror of https://github.com/go-kratos/kratos.git synced 2026-05-16 09:48:28 +02:00

add validator middleware

This commit is contained in:
chenzhihui
2021-02-18 20:14:36 +08:00
parent c7827ff701
commit ad97e72c15
+26
View File
@@ -0,0 +1,26 @@
package validate
import (
"context"
"github.com/go-kratos/kratos/v2/errors"
"github.com/go-kratos/kratos/v2/middleware"
)
type validator interface {
Validate() error
}
// Validator is a validator middleware.
func Validator() middleware.Middleware {
return func(handler middleware.Handler) middleware.Handler {
return func(ctx context.Context, req interface{}) (reply interface{}, err error) {
if v, ok := req.(validator); ok {
if err := v.Validate(); err != nil {
return nil, errors.InvalidArgument("Validator", err.Error())
}
}
return handler(ctx, req)
}
}
}