fix: 代码自动生成选择业务库自动迁移代码生成异常 (flipped-aurora#1848)

lbbniu · web-flow · commit 6fedb9773bb5 · 2024-08-12T15:08:17.000+08:00
diff --git a/server/utils/ast/package_initialize_gorm.go b/server/utils/ast/package_initialize_gorm.go
@@ -142,7 +142,7 @@ func (a *PackageInitializeGorm) addDbVar(astBody *ast.BlockStmt) {
 	for i := range astBody.List {
 		if assignStmt, ok := astBody.List[i].(*ast.AssignStmt); ok {
 			if ident, ok := assignStmt.Lhs[0].(*ast.Ident); ok {
-				if ident.Name == "db" || ident.Name == a.Business+"Db" {
+				if (a.Business == "" && ident.Name == "db") || ident.Name == a.Business+"Db" {
 					return
 				}
 			}

Original file line number	Diff line number	Diff line change
`@@ -142,7 +142,7 @@ func (a PackageInitializeGorm) addDbVar(astBody ast.BlockStmt) {`
`142`	`142`	`for i := range astBody.List {`
`143`	`143`	`if assignStmt, ok := astBody.List[i].(*ast.AssignStmt); ok {`
`144`	`144`	`if ident, ok := assignStmt.Lhs[0].(*ast.Ident); ok {`
`145`		`- if ident.Name == "db" \|\| ident.Name == a.Business+"Db" {`
	`145`	`+ if (a.Business == "" && ident.Name == "db") \|\| ident.Name == a.Business+"Db" {`
`146`	`146`	`return`
`147`	`147`	`}`
`148`	`148`	`}`