You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.7 KiB
Go
59 lines
1.7 KiB
Go
package gogen
|
|
|
|
import (
|
|
_ "embed"
|
|
"fmt"
|
|
"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/format"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
|
|
"github.com/zeromicro/go-zero/tools/goctl/vars"
|
|
)
|
|
|
|
const contextFilename = "service_context"
|
|
|
|
//go:embed svc.tpl
|
|
var contextTemplate string
|
|
|
|
func genServiceContext(dir, rootPkg string, cfg *config.Config, api *spec.ApiSpec) error {
|
|
filename, err := format.FileNamingFormat(cfg.NamingFormat, contextFilename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var middlewareStr string
|
|
var middlewareAssignment string
|
|
middlewares := getMiddleware(api)
|
|
|
|
for _, item := range middlewares {
|
|
middlewareStr += fmt.Sprintf("%s rest.Middleware\n", item)
|
|
name := strings.TrimSuffix(item, "Middleware") + "Middleware"
|
|
middlewareAssignment += fmt.Sprintf("%s: %s,\n", item,
|
|
fmt.Sprintf("middleware.New%s().%s", strings.Title(name), "Handle"))
|
|
}
|
|
|
|
configImport := "\"" + pathx.JoinPackages(rootPkg, configDir) + "\""
|
|
if len(middlewareStr) > 0 {
|
|
configImport += "\n\t\"" + pathx.JoinPackages(rootPkg, middlewareDir) + "\""
|
|
configImport += fmt.Sprintf("\n\t\"%s/rest\"", vars.ProjectOpenSourceURL)
|
|
}
|
|
|
|
return genFile(fileGenConfig{
|
|
dir: dir,
|
|
subdir: contextDir,
|
|
filename: filename + ".go",
|
|
templateName: "contextTemplate",
|
|
category: category,
|
|
templateFile: contextTemplateFile,
|
|
builtinTemplate: contextTemplate,
|
|
data: map[string]string{
|
|
"configImport": configImport,
|
|
"config": "config.Config",
|
|
"middleware": middlewareStr,
|
|
"middlewareAssignment": middlewareAssignment,
|
|
},
|
|
})
|
|
}
|