fix:replace the method of abandonment (#1075)

* fix:replace the method of abandonment
This commit is contained in:
songzhibin97
2022-05-19 11:10:05 +08:00
committed by GitHub
parent 02ffe71d80
commit e4709c599b
4 changed files with 28 additions and 20 deletions

View File

@@ -10,6 +10,8 @@ import (
"go/format"
"go/parser"
"go/token"
"golang.org/x/text/cases"
"golang.org/x/text/language"
"io/ioutil"
"log"
"os"
@@ -48,6 +50,7 @@ type autoPackage struct {
var (
packageInjectionMap map[string]astInjectionMeta
injectionPaths []injectionMeta
caser = cases.Title(language.English)
)
func Init(Package string) {
@@ -619,7 +622,7 @@ func (autoCodeService *AutoCodeService) CreatePackageTemp(packageName string) er
// 创建完成后在对应的位置插入结构代码
for _, v := range pendingTemp {
meta := packageInjectionMap[v.name]
if err := ImportReference(meta.path, fmt.Sprintf(meta.importCodeF, v.name, packageName), fmt.Sprintf(meta.structNameF, strings.Title(packageName)), fmt.Sprintf(meta.packageNameF, packageName), meta.groupName); err != nil {
if err := ImportReference(meta.path, fmt.Sprintf(meta.importCodeF, v.name, packageName), fmt.Sprintf(meta.structNameF, caser.String(packageName)), fmt.Sprintf(meta.packageNameF, packageName), meta.groupName); err != nil {
return err
}
}
@@ -750,7 +753,7 @@ func (vi *Visitor) addFuncBodyVar(funDecl *ast.FuncDecl) ast.Visitor {
},
},
Sel: &ast.Ident{
Name: strings.Title(vi.PackageName),
Name: caser.String(vi.PackageName),
},
},
},