feat:filter_test_suffix

songzhibin97 · songzhibin97 · commit adea9236d2fc · 2022-07-07T19:10:10.000+08:00
diff --git a/server/api/v1/system/sys_auto_code.go b/server/api/v1/system/sys_auto_code.go
@@ -37,7 +37,7 @@ func (autoApi *AutoCodeApi) PreviewTemp(c *gin.Context) {
 		response.FailWithMessage(err.Error(), c)
 		return
 	}
-	a.KeyWord() // 处理go关键字
+	a.Pretreatment() // 处理go关键字
 	a.PackageT = caser.String(a.Package)
 	autoCode, err := autoCodeService.PreviewTemp(a)
 	if err != nil {
@@ -64,7 +64,7 @@ func (autoApi *AutoCodeApi) CreateTemp(c *gin.Context) {
 		response.FailWithMessage(err.Error(), c)
 		return
 	}
-	a.KeyWord() // 处理go关键字
+	a.Pretreatment()
 	var apiIds []uint
 	if a.AutoCreateApiToSql {
 		if ids, err := autoCodeService.AutoCreateApi(&a); err != nil {
diff --git a/server/model/system/sys_auto_code.go b/server/model/system/sys_auto_code.go
@@ -4,6 +4,7 @@ import (
 	"errors"
 	"github.com/flipped-aurora/gin-vue-admin/server/global"
 	"go/token"
+	"strings"
 )
 
 // AutoCodeStruct 初始版本自动化代码工具
@@ -23,6 +24,11 @@ type AutoCodeStruct struct {
 	PackageT           string   `json:"-"`
 }
 
+func (a *AutoCodeStruct) Pretreatment() {
+	a.KeyWord()
+	a.SuffixTest()
+}
+
 // KeyWord 是go关键字的处理加上 _ ，防止编译报错
 // Author [SliverHorn](https://github.com/SliverHorn)
 func (a *AutoCodeStruct) KeyWord() {
@@ -31,6 +37,14 @@ func (a *AutoCodeStruct) KeyWord() {
 	}
 }
 
+// SuffixTest 处理_test 后缀
+// Author [SliverHorn](https://github.com/SliverHorn)
+func (a *AutoCodeStruct) SuffixTest() {
+	if strings.HasSuffix(a.Package, "test") {
+		a.Package = a.Package + "_"
+	}
+}
+
 type Field struct {
 	FieldName       string `json:"fieldName"`       // Field名
 	FieldDesc       string `json:"fieldDesc"`       // 中文名
