blackteay/gin
3
0
Fork 0
mirror of https://github.com/gin-gonic/gin.git synced 2025-04-06 03:57:46 +08:00
Code Issues Projects Releases Wiki Activity

feature: add ability to handle the binder errors

Browse Source
This commit is contained in:
abdulrahman 2024-01-01 12:52:53 +04:00
parent 0e0ea62a94
commit 14bae4aef3
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
auto_binder.go
View File

@ -6,6 +6,13 @@ import (
"reflect" "reflect"
) )
var (
defaultAutoBinderErrorHandler = func(ctx *Context, err error) {
ctx.Error(err)
ctx.Abort()
}
)
type binderType func(obj any) error type binderType func(obj any) error
func isFunc(obj any) bool { func isFunc(obj any) bool {
@ -75,8 +82,10 @@ func callHandler(rt reflect.Type, rv reflect.Value, ctx *Context, binder binderT
// //
// Example: func MyGetHandler(ctx *gin.Context, request *MyRequest) {} // Example: func MyGetHandler(ctx *gin.Context, request *MyRequest) {}
// //
// engine.GET("/endpoint", gin.AutoBinder(MyGetHandler)) // engine.GET("/endpoint", gin.AutoBinder(MyGetHandler)) and you can handel the errors by passing a handler
func AutoBinder(handler any) HandlerFunc { //
// engine.GET("/endpoint", gin.AutoBinder(MyGetHandler, func(ctx *gin.Context, err error) {}))
func AutoBinder(handler any, errorHandler ...func(*Context, error)) HandlerFunc {
rt := reflect.TypeOf(handler) rt := reflect.TypeOf(handler)
if rt.Kind() != reflect.Func { if rt.Kind() != reflect.Func {
@ -88,13 +97,18 @@ func AutoBinder(handler any) HandlerFunc {
} }
return func(ctx *Context) { return func(ctx *Context) {
selectedErrorHandler := defaultAutoBinderErrorHandler
if len(errorHandler) > 0 && errorHandler[0] != nil {
selectedErrorHandler = errorHandler[0]
}
rt := reflect.TypeOf(handler) rt := reflect.TypeOf(handler)
rv := reflect.ValueOf(handler) rv := reflect.ValueOf(handler)
if err := callHandler(rt, rv, ctx, func(obj any) error { if err := callHandler(rt, rv, ctx, func(obj any) error {
return ctx.ShouldBind(obj) return ctx.ShouldBind(obj)
}); err != nil { }); err != nil {
ctx.Error(err) selectedErrorHandler(ctx, err)
} }
} }
} }