lbsadmin/bee
1
0
Fork 0
mirror of https://github.com/beego/bee.git synced 2024-11-26 06:21:34 +00:00
Code Issues Releases Activity

添加自动话路由容错

Browse Source
This commit is contained in:
候胜斌 2018-10-26 17:44:17 +08:00
parent 10bb0454f6
commit a56257aaaf
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
generate/swaggergen/g_docs.go
View File

@ -36,7 +36,7 @@ import (
"github.com/astaxie/beego/swagger" "github.com/astaxie/beego/swagger"
"github.com/astaxie/beego/utils" "github.com/astaxie/beego/utils"
beeLogger "github.com/beego/bee/logger" "github.com/beego/bee/logger"
bu "github.com/beego/bee/utils" bu "github.com/beego/bee/utils"
) )
@ -519,6 +519,17 @@ func peekNextSplitString(ss string) (s string, spacePos int) {
} }
return return
} }
/**
Comment format with a single space as the semantic segmentation,
single comma for the array information division
*/
func formatComment(comment string) string {
spaceReg, _ := regexp.Compile("[ \t]+")
commaReg, _ := regexp.Compile(",[ \t]+")
comment = spaceReg.ReplaceAllString(comment, " ")
comment = commaReg.ReplaceAllString(comment, ",")
return comment
}
// parse the func comments // parse the func comments
func parserComments(f *ast.FuncDecl, controllerName, pkgpath string) error { func parserComments(f *ast.FuncDecl, controllerName, pkgpath string) error {
@ -535,6 +546,7 @@ func parserComments(f *ast.FuncDecl, controllerName, pkgpath string) error {
for _, c := range comments.List { for _, c := range comments.List {
t := strings.TrimSpace(strings.TrimLeft(c.Text, "//")) t := strings.TrimSpace(strings.TrimLeft(c.Text, "//"))
if strings.HasPrefix(t, "@router") { if strings.HasPrefix(t, "@router") {
t = formatComment(t)
elements := strings.TrimSpace(t[len("@router"):]) elements := strings.TrimSpace(t[len("@router"):])
e1 := strings.SplitN(elements, " ", 2) e1 := strings.SplitN(elements, " ", 2)
if len(e1) < 1 { if len(e1) < 1 {