optimize gin template file

alimy · Mar 6, 2020 · c7b25e9 · c7b25e9
1 parent 543e082
commit c7b25e9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 13 deletions.
diff --git a/internal/generator/templates/gin_iface.tmpl b/internal/generator/templates/gin_iface.tmpl
@@ -11,21 +11,17 @@ type {{.TypeName}} interface {
 {{if .IsNeedChain}}// Chain provide handlers chain for gin
     Chain() gin.HandlersChain
 {{end}}
-{{range .Fields}}{{if notEmptyStr .Comment }}// {{.Comment}}{{end}}
-    {{.MethodName}}(*gin.Context)
-{{end}}
+{{range .Fields}}{{if notEmptyStr .Comment }}    // {{.Comment}}
+    {{.MethodName}}(*gin.Context){{else}}    {{.MethodName}}(*gin.Context){{end}}{{end}}
 }
 
 // Register{{.TypeName}}Servant register {{.TypeName}} servant to gin
 func Register{{.TypeName}}Servant(e *gin.Engine, s {{.TypeName}}) {
-{{if notEmptyStr .Group }}router := e.Group({{.Group}}){{else}}router := e{{end}}
-{{if .IsNeedChain}}// use chain for router
+{{if notEmptyStr .Group }}    router := e.Group({{.Group}}){{else}}    router := e{{end}}
+{{if .IsNeedChain}}    // use chain for router
     middlewares := s.Chain()
     router.Use(middlewares...)
 {{end}}
-
     // register routes info to router
-{{range .Fields}}
-    {{if notHttpAny .HttpMethod }}router.Handle({{.HttpMethod}}, "{{.Path}}", s.{{.MethodName}}){{else}}router.Any("{{.Path}}", s.{{.MethodName}}){{end}}
-{{end}}
+{{range .Fields}}{{if notHttpAny .HttpMethod }}router.Handle({{.HttpMethod}}, "{{.Path}}", s.{{.MethodName}}){{else}}router.Any("{{.Path}}", s.{{.MethodName}}){{end}}{{end}}
 }
diff --git a/internal/generator/templates_gen.go b/internal/generator/templates_gen.go