feat: 调整自动化代码逻辑，重复创建的菜单和api需要用户手动取消自动创建 (#1798)

* feat: 调整自动化代码逻辑，重复创建的菜单和api需要用户手动取消自动创建 --------- Co-authored-by: wall-js <[email protected]>
flipped-aurora · Jun 27, 2024 · 45f723c · 45f723c
1 parent 640f44f
commit 45f723c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/server/api/v1/system/sys_auto_code.go b/server/api/v1/system/sys_auto_code.go
@@ -64,19 +64,20 @@ func (autoApi *AutoCodeApi) CreateTemp(c *gin.Context) {
 	var menuId uint
 	if a.AutoCreateApiToSql {
 		if ids, err := autoCodeService.AutoCreateApi(&a); err != nil {
-			global.GVA_LOG.Error("自动化创建失败!请自行清空垃圾数据!", zap.Error(err))
+			global.GVA_LOG.Error("自动化创建API失败!", zap.Error(err))
 			c.Writer.Header().Add("success", "false")
-			c.Writer.Header().Add("msg", url.QueryEscape("自动化创建失败!请自行清空垃圾数据!"))
+			c.Writer.Header().Add("msg", url.QueryEscape("自动化创建失败!请自行清空垃圾数据或取消自动创建API!"))
 			return
 		} else {
 			apiIds = ids
 		}
 	}
 	if a.AutoCreateMenuToSql {
 		if id, err := autoCodeService.AutoCreateMenu(&a); err != nil {
-			global.GVA_LOG.Error("自动化创建失败!请自行清空垃圾数据!", zap.Error(err))
+			global.GVA_LOG.Error("自动化创建菜单失败!", zap.Error(err))
 			c.Writer.Header().Add("success", "false")
-			c.Writer.Header().Add("msg", url.QueryEscape("自动化创建失败!请自行清空垃圾数据!"))
+			c.Writer.Header().Add("msg", url.QueryEscape("自动化创建失败!请自行清空垃圾数据或取消自动创建菜单!"))
+			return
 		} else {
 			menuId = id
 		}

diff --git a/server/service/system/sys_auto_code.go b/server/service/system/sys_auto_code.go
@@ -5,7 +5,6 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
-	"github.com/mholt/archiver/v4"
 	"io"
 	"mime/multipart"
 	"os"
@@ -14,6 +13,8 @@ import (
 	"strings"
 	"text/template"
 
+	"github.com/mholt/archiver/v4"
+
 	ast2 "github.com/flipped-aurora/gin-vue-admin/server/utils/ast"
 
 	"github.com/flipped-aurora/gin-vue-admin/server/resource/autocode_template/subcontract"
@@ -468,7 +469,11 @@ func (autoCodeService *AutoCodeService) AutoCreateApi(a *system.AutoCodeStruct)
 	err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
 		for _, v := range apiList {
 			var api system.SysApi
-			if errors.Is(tx.Where("path = ? AND method = ?", v.Path, v.Method).First(&api).Error, gorm.ErrRecordNotFound) {
+			findErr := tx.Where("path = ? AND method = ?", v.Path, v.Method).First(&api).Error
+			if findErr == nil {
+				return errors.New("存在相同的API，请关闭自动创建API功能")
+			}
+			if errors.Is(findErr, gorm.ErrRecordNotFound) {
 				if err = tx.Create(&v).Error; err != nil { // 遇到错误时回滚事务
 					return err
 				} else {