53 lines
1.6 KiB
Go
53 lines
1.6 KiB
Go
|
/*
|
|||
|
|
* @desc:路由处理
|
|||
|
|
* @company:云南奇讯科技有限公司
|
|||
|
|
* @Author: yixiaohu<yxh669@qq.com>
|
|||
|
|
* @Date: 2022/11/16 11:09
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
package libRouter
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|||
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
|||
|
|
"github.com/gogf/gf/v2/text/gregex"
|
|||
|
|
"reflect"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// RouterAutoBindBefore 收集需要被绑定的不验证用户登录状态的控制器,自动绑定
|
|||
|
|
// 路由的方法命名规则必须为：BeforeBindXXXController
|
|||
|
|
func RouterAutoBindBefore(ctx context.Context, R interface{}, group *ghttp.RouterGroup) (err error) {
|
|||
|
|
return bind(ctx, R, group, "before")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RouterAutoBind 收集需要被绑定的控制器,自动绑定
|
|||
|
|
// 路由的方法命名规则必须为：BindXXXController
|
|||
|
|
func RouterAutoBind(ctx context.Context, R interface{}, group *ghttp.RouterGroup) (err error) {
|
|||
|
|
return bind(ctx, R, group)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func bind(ctx context.Context, R interface{}, group *ghttp.RouterGroup, option ...string) (err error) {
|
|||
|
|
var rule string
|
|||
|
|
if len(option) > 0 && option[0] == "before" {
|
|||
|
|
rule = `^BeforeBind(.+)Controller$`
|
|||
|
|
} else {
|
|||
|
|
rule = `^Bind(.+)Controller$`
|
|||
|
|
}
|
|||
|
|
//TypeOf会返回目标数据的类型，比如int/float/struct/指针等
|
|||
|
|
typ := reflect.TypeOf(R)
|
|||
|
|
//ValueOf返回目标数据的的值
|
|||
|
|
val := reflect.ValueOf(R)
|
|||
|
|
if val.Elem().Kind() != reflect.Struct {
|
|||
|
|
err = gerror.New("expect struct but a " + val.Elem().Kind().String())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
for i := 0; i < typ.NumMethod(); i++ {
|
|||
|
|
if match := gregex.IsMatchString(rule, typ.Method(i).Name); match {
|
|||
|
|
//调用绑定方法
|
|||
|
|
val.Method(i).Call([]reflect.Value{reflect.ValueOf(ctx), reflect.ValueOf(group)})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|