server/beego
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Optimize AddAutoPrefix: only register one router if it's not case sensetive

Browse Source
This commit is contained in:
Ming Deng 2021-04-19 22:12:34 +08:00
parent 3d511fd30b
commit d6f80689a0
1 changed files with 23 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
server/web/router.go
View File

@ -732,22 +732,32 @@ func (p *ControllerRegister) AddAutoPrefix(prefix string, c ControllerInterface)
ct := reflect.Indirect(reflectVal).Type()
controllerName := strings.TrimSuffix(ct.Name(), "Controller")
for i := 0; i < rt.NumMethod(); i++ {
if !utils.InSlice(rt.Method(i).Name, exceptMethod) {
pattern := path.Join(prefix, strings.ToLower(controllerName), strings.ToLower(rt.Method(i).Name), "*")
patternInit := path.Join(prefix, controllerName, rt.Method(i).Name, "*")
patternFix := path.Join(prefix, strings.ToLower(controllerName), strings.ToLower(rt.Method(i).Name))
patternFixInit := path.Join(prefix, controllerName, rt.Method(i).Name)
methodName := rt.Method(i).Name
if !utils.InSlice(methodName, exceptMethod) {
p.addAutoPrefixMethod(prefix, controllerName, methodName, ct)
}
}
}
route := p.createBeegoRouter(ct, pattern)
route.methods = map[string]string{"*": rt.Method(i).Name}
func (p *ControllerRegister) addAutoPrefixMethod(prefix, controllerName, methodName string, ctrl reflect.Type) {
pattern := path.Join(prefix, strings.ToLower(controllerName), strings.ToLower(methodName), "*")
patternInit := path.Join(prefix, controllerName, methodName, "*")
patternFix := path.Join(prefix, strings.ToLower(controllerName), strings.ToLower(methodName))
patternFixInit := path.Join(prefix, controllerName, methodName)
route := p.createBeegoRouter(ctrl, pattern)
route.methods = map[string]string{"*": methodName}
for m := range HTTPMETHOD {
p.addToRouter(m, pattern, route)
// only case sensitive, we add three more routes
if p.cfg.RouterCaseSensitive {
p.addToRouter(m, patternInit, route)
p.addToRouter(m, patternFix, route)
p.addToRouter(m, patternFixInit, route)
}
}
}
}
// InsertFilter Add a FilterFunc with pattern rule and action constant.