增加回滚不删表按钮防止误删数据库重要数据

Author: piexlmax

server/api/v1/system/sys_auto_code_history.go

@@ -58,11 +58,11 @@ func (a *AutoCodeHistoryApi) Delete(c *gin.Context) {
 // @Security ApiKeyAuth
 // @accept application/json
 // @Produce application/json
-// @Param data body request.GetById true "请求参数"
+// @Param data body systemReq.RollBack true "请求参数"
 // @Success 200 {object} response.Response{msg=string} "回滚自动生成代码"
 // @Router /autoCode/rollback [post]
 func (a *AutoCodeHistoryApi) RollBack(c *gin.Context) {
-	var info request.GetById
+	var info systemReq.RollBack
 	_ = c.ShouldBindJSON(&info)
 	if err := autoCodeHistoryService.RollBack(&info); err != nil {
 		response.FailWithMessage(err.Error(), c)

server/model/system/request/sys_auto_history.go

@@ -5,3 +5,9 @@ import "github.com/flipped-aurora/gin-vue-admin/server/model/common/request"
 type SysAutoHistory struct {
 	request.PageInfo
 }
+
+// GetById Find by id structure
+type RollBack struct {
+	ID int `json:"id" form:"id"` // 主键ID
+	DeleteTable bool `json:"deleteTable" form:"deleteTable"` // 是否删除表
+}

server/service/system/sys_autocode_history.go

@@ -3,6 +3,7 @@ package system
 import (
 	"errors"
 	"fmt"
+	systemReq "github.com/flipped-aurora/gin-vue-admin/server/model/system/request"
 	"path/filepath"
 	"strings"
 	"time"
@@ -60,9 +61,9 @@ func (autoCodeHistoryService *AutoCodeHistoryService) Repeat(structName string,
 // RollBack 回滚
 // Author [SliverHorn](https://github.com/SliverHorn)
 // Author [songzhibin97](https://github.com/songzhibin97)
-func (autoCodeHistoryService *AutoCodeHistoryService) RollBack(info *request.GetById) error {
+func (autoCodeHistoryService *AutoCodeHistoryService) RollBack(info *systemReq.RollBack) error {
 	md := system.SysAutoCodeHistory{}
-	if err := global.GVA_DB.Where("id = ?", info.Uint()).First(&md).Error; err != nil {
+	if err := global.GVA_DB.Where("id = ?", info.ID).First(&md).Error; err != nil {
 		return err
 	}
 	// 清除API表
@@ -71,8 +72,10 @@ func (autoCodeHistoryService *AutoCodeHistoryService) RollBack(info *request.Get
 		global.GVA_LOG.Error("ClearTag DeleteApiByIds:", zap.Error(err))
 	}
 	// 删除表
-	if err = AutoCodeServiceApp.DropTable(md.TableName); err != nil {
-		global.GVA_LOG.Error("ClearTag DropTable:", zap.Error(err))
+	if info.DeleteTable {
+		if err = AutoCodeServiceApp.DropTable(md.TableName); err != nil {
+			global.GVA_LOG.Error("ClearTag DropTable:", zap.Error(err))
+		}
 	}
 	// 删除文件
 

web/src/view/systemTools/autoCodeAdmin/index.vue

@@ -36,10 +36,11 @@
             </el-tag>
           </template>
         </el-table-column>
-        <el-table-column align="left" label="操作" min-width="180">
+        <el-table-column align="left" label="操作" min-width="240">
           <template #default="scope">
             <div>
-              <el-button size="small" type="text" :disabled="scope.row.flag === 1" @click="rollbackFunc(scope.row)">回滚</el-button>
+              <el-button size="small" type="text" :disabled="scope.row.flag === 1" @click="rollbackFunc(scope.row,true)">回滚(删表)</el-button>
+              <el-button size="small" type="text" :disabled="scope.row.flag === 1" @click="rollbackFunc(scope.row,false)">回滚(不删表)</el-button>
               <el-button size="small" type="text" @click="goAutoCode(scope.row)">复用</el-button>
               <el-button size="small" type="text" @click="deleteRow(scope.row)">删除</el-button>
             </div>
@@ -120,13 +121,13 @@ const deleteRow = async(row) => {
     }
   })
 }
-const rollbackFunc = async(row) => {
-  ElMessageBox.confirm('此操作将删除自动创建的文件和api, 是否继续?', '提示', {
+const rollbackFunc = async(row, flag) => {
+  ElMessageBox.confirm(`此操作将删除自动创建的文件和api${flag ? '（包含数据库表！）' : ''}, 是否继续?`, '提示', {
     confirmButtonText: '确定',
     cancelButtonText: '取消',
     type: 'warning'
   }).then(async() => {
-    const res = await rollback({ id: Number(row.ID) })
+    const res = await rollback({ id: Number(row.ID), deleteTable: flag })
     if (res.code === 0) {
       ElMessage.success('回滚成功')
       getTableData()