fix: 代码生成器 结构体简称发生的关键字冲突 flipped-aurora#1089

SliverHorn · SliverHorn · commit c4f5919d7831 · 2022-05-27T13:50:27.000+08:00
diff --git a/server/api/v1/system/sys_auto_code.go b/server/api/v1/system/sys_auto_code.go
@@ -38,7 +38,7 @@ func (autoApi *AutoCodeApi) PreviewTemp(c *gin.Context) {
 		response.FailWithMessage(err.Error(), c)
 		return
 	}
-
+	a.KeyWord() // 处理go关键字
 	a.PackageT = caser.String(a.Package)
 	autoCode, err := autoCodeService.PreviewTemp(a)
 	if err != nil {
@@ -65,6 +65,7 @@ func (autoApi *AutoCodeApi) CreateTemp(c *gin.Context) {
 		response.FailWithMessage(err.Error(), c)
 		return
 	}
+	a.KeyWord() // 处理go关键字
 	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
@@ -3,6 +3,7 @@ package system
 import (
 	"errors"
 	"github.com/flipped-aurora/gin-vue-admin/server/global"
+	"go/token"
 )
 
 // AutoCodeStruct 初始版本自动化代码工具
@@ -21,6 +22,14 @@ type AutoCodeStruct struct {
 	PackageT           string   `json:"-"`
 }
 
+// KeyWord 是go关键字的处理加上 _ ，防止编译报错
+// Author [SliverHorn](https://github.com/SliverHorn)
+func (a *AutoCodeStruct) KeyWord() {
+	if token.IsKeyword(a.Abbreviation) {
+		a.Abbreviation = "_" + a.Abbreviation
+	}
+}
+
 type Field struct {
 	FieldName       string `json:"fieldName"`       // Field名
 	FieldDesc       string `json:"fieldDesc"`       // 中文名
