mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-02-03 00:38:40 +08:00
124 lines
3.3 KiB
Go
124 lines
3.3 KiB
Go
package gogen
|
|
|
|
import (
|
|
_ "embed"
|
|
"fmt"
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/zeromicro/go-zero/tools/goctl/api/spec"
|
|
"github.com/zeromicro/go-zero/tools/goctl/config"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/format"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
|
|
)
|
|
|
|
const defaultLogicPackage = "logic"
|
|
|
|
//go:embed handler.tpl
|
|
var handlerTemplate string
|
|
|
|
func genHandler(dir, rootPkg string, cfg *config.Config, group spec.Group, route spec.Route) error {
|
|
handler := getHandlerName(route)
|
|
handlerPath := getHandlerFolderPath(group, route)
|
|
pkgName := handlerPath[strings.LastIndex(handlerPath, "/")+1:]
|
|
logicName := defaultLogicPackage
|
|
if handlerPath != handlerDir {
|
|
handler = strings.Title(handler)
|
|
logicName = pkgName
|
|
}
|
|
filename, err := format.FileNamingFormat(cfg.NamingFormat, handler)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return genFile(fileGenConfig{
|
|
dir: dir,
|
|
subdir: getHandlerFolderPath(group, route),
|
|
filename: filename + ".go",
|
|
templateName: "handlerTemplate",
|
|
category: category,
|
|
templateFile: handlerTemplateFile,
|
|
builtinTemplate: handlerTemplate,
|
|
data: map[string]any{
|
|
"PkgName": pkgName,
|
|
"ImportPackages": genHandlerImports(group, route, rootPkg),
|
|
"HandlerName": handler,
|
|
"RequestType": util.Title(route.RequestTypeName()),
|
|
"LogicName": logicName,
|
|
"LogicType": strings.Title(getLogicName(route)),
|
|
"Call": strings.Title(strings.TrimSuffix(handler, "Handler")),
|
|
"HasResp": len(route.ResponseTypeName()) > 0,
|
|
"HasRequest": len(route.RequestTypeName()) > 0,
|
|
"HasDoc": len(route.JoinedDoc()) > 0,
|
|
"Doc": getDoc(route.JoinedDoc()),
|
|
},
|
|
})
|
|
}
|
|
|
|
func genHandlers(dir, rootPkg string, cfg *config.Config, api *spec.ApiSpec) error {
|
|
for _, group := range api.Service.Groups {
|
|
for _, route := range group.Routes {
|
|
if err := genHandler(dir, rootPkg, cfg, group, route); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func genHandlerImports(group spec.Group, route spec.Route, parentPkg string) string {
|
|
imports := []string{
|
|
fmt.Sprintf("\"%s\"", pathx.JoinPackages(parentPkg, getLogicFolderPath(group, route))),
|
|
fmt.Sprintf("\"%s\"", pathx.JoinPackages(parentPkg, contextDir)),
|
|
}
|
|
if len(route.RequestTypeName()) > 0 {
|
|
imports = append(imports, fmt.Sprintf("\"%s\"\n", pathx.JoinPackages(parentPkg, typesDir)))
|
|
}
|
|
|
|
return strings.Join(imports, "\n\t")
|
|
}
|
|
|
|
func getHandlerBaseName(route spec.Route) (string, error) {
|
|
handler := route.Handler
|
|
handler = strings.TrimSpace(handler)
|
|
handler = strings.TrimSuffix(handler, "handler")
|
|
handler = strings.TrimSuffix(handler, "Handler")
|
|
|
|
return handler, nil
|
|
}
|
|
|
|
func getHandlerFolderPath(group spec.Group, route spec.Route) string {
|
|
folder := route.GetAnnotation(groupProperty)
|
|
if len(folder) == 0 {
|
|
folder = group.GetAnnotation(groupProperty)
|
|
if len(folder) == 0 {
|
|
return handlerDir
|
|
}
|
|
}
|
|
|
|
folder = strings.TrimPrefix(folder, "/")
|
|
folder = strings.TrimSuffix(folder, "/")
|
|
|
|
return path.Join(handlerDir, folder)
|
|
}
|
|
|
|
func getHandlerName(route spec.Route) string {
|
|
handler, err := getHandlerBaseName(route)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return handler + "Handler"
|
|
}
|
|
|
|
func getLogicName(route spec.Route) string {
|
|
handler, err := getHandlerBaseName(route)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return handler + "Logic"
|
|
}
|