* optimize performance

* rename

* rename

* revert
This commit is contained in:
kingxt
2021-05-29 23:01:02 +08:00
committed by GitHub
parent 3022f93b6d
commit 546fcd8bab
6 changed files with 25 additions and 44 deletions

View File

@@ -39,7 +39,7 @@ func main() {
}
`
func genMain(dir string, cfg *config.Config, api *spec.ApiSpec) error {
func genMain(dir, rootPkg string, cfg *config.Config, api *spec.ApiSpec) error {
name := strings.ToLower(api.Service.Name)
if strings.HasSuffix(name, "-api") {
name = strings.ReplaceAll(name, "-api", "")
@@ -49,11 +49,6 @@ func genMain(dir string, cfg *config.Config, api *spec.ApiSpec) error {
return err
}
parentPkg, err := getParentPackage(dir)
if err != nil {
return err
}
return genFile(fileGenConfig{
dir: dir,
subdir: "",
@@ -63,7 +58,7 @@ func genMain(dir string, cfg *config.Config, api *spec.ApiSpec) error {
templateFile: mainTemplateFile,
builtinTemplate: mainTemplate,
data: map[string]string{
"importPackages": genMainImports(parentPkg),
"importPackages": genMainImports(rootPkg),
"serviceName": api.Service.Name,
},
})