✨ 新增保存菜单接口

api/admin/menu/save.go

@@ -0,0 +1,24 @@
+package menu
+
+import (
+	"github.com/gin-gonic/gin"
+	menuParam "wechat-robot/model/param/menu"
+	"wechat-robot/pkg/response"
+	"wechat-robot/service/menu"
+)
+
+// Save
+// @description: 保存菜单
+// @param ctx
+func Save(ctx *gin.Context) {
+	var p menuParam.Save
+	if err := ctx.ShouldBind(&p); err != nil {
+		response.New(ctx).SetError(err).Fail()
+		return
+	}
+	if err := menu.Save(p); err != nil {
+		response.New(ctx).SetError(err).Fail()
+		return
+	}
+	response.New(ctx).Success()
+}

internal/orm/sort.go

@@ -0,0 +1,148 @@
+package orm
+
+import (
+	"fmt"
+	"gitee.ltd/lxh/logger/log"
+	"gorm.io/gorm"
+	"slices"
+)
+
+// @title checkHasDeletedAt
+// @description 检查指定表是否有id_del字段
+// @param tx *gorm.DB "已开启的事务对象"
+// @param tableName string "表名"
+// @return bool "是否包含"
+func checkHasDeletedAt(tx *gorm.DB, tableName string) bool {
+	var columns []string
+	// SQL语句
+	sql := fmt.Sprintf("select COLUMN_NAME from information_schema.COLUMNS where table_name = '%s'", tableName)
+	err := tx.Raw(sql).Scan(&columns).Error
+	if err != nil {
+		log.Errorf("查询表字段失败: %v", err.Error())
+		return false
+	}
+	return slices.Contains(columns, "id_del")
+}
+
+// @title checkHasUpdatedAt
+// @description 检查指定表是否有updated_at字段
+// @param tx *gorm.DB "已开启的事务对象"
+// @param tableName string "表名"
+// @return bool "是否包含"
+func checkHasUpdatedAt(tx *gorm.DB, tableName string) bool {
+	var columns []string
+	// SQL语句
+	sql := fmt.Sprintf("select COLUMN_NAME from information_schema.COLUMNS where table_name = '%s'", tableName)
+	err := tx.Raw(sql).Scan(&columns).Error
+	if err != nil {
+		log.Errorf("查询表字段失败: %v", err.Error())
+		return false
+	}
+	return slices.Contains(columns, "updated_at")
+}
+
+// UpdateSortBefore
+// @description 更新之前处理序号
+// @param tx *gorm.DB "已开启的事务对象"
+// @param model any "模型对象"
+// @return error "错误信息"
+func UpdateSortBefore(tx *gorm.DB, tableName, id string, sort int, param string) (err error) {
+	// 查出原来的排序号
+	var oldSort int
+	err = tx.Table(tableName).Select("`sort`").Where("id = ?", id).Scan(&oldSort).Error
+	if err != nil {
+		log.Errorf("查询老数据失败: %v", err.Error())
+		return
+	}
+	// 如果相等，啥都不干
+	if oldSort == sort {
+		return nil
+	}
+
+	// 查询是否包含 id_del 字段
+	hasDeletedAt := checkHasDeletedAt(tx, tableName)
+
+	// 处理排序
+	// 如果老的排序号小于新的，(老, 新]之间的排序号都要-1
+	// 如果老的大于新的，[老, 新)排序号-1
+	if oldSort < sort {
+		// 老的小于新的，[老, 新) + 1
+		sel := tx.Table(tableName).
+			Where("sort <= ? AND sort > ?", sort, oldSort)
+		if hasDeletedAt {
+			sel.Where("id_del = 0")
+		}
+		if param != "" {
+			sel.Where(param) // 自定义条件
+		}
+		err = sel.Update("sort", gorm.Expr("sort - 1")).Error
+	} else {
+		// 老的大于新的，[新, 老) + 1
+		sel := tx.Table(tableName).
+			Where("sort >= ? AND sort < ?", sort, oldSort)
+		if hasDeletedAt {
+			sel.Where("id_del = 0")
+		}
+		if param != "" {
+			sel.Where(param) // 自定义条件
+		}
+		err = sel.Update("sort", gorm.Expr("sort + 1")).Error
+	}
+	return
+}
+
+// CreateSortBefore
+// @description 新建之前处理序号
+// @param tx *gorm.DB "已开启的事务对象"
+// @param model any "模型对象"
+// @return error "错误信息"
+func CreateSortBefore(tx *gorm.DB, tableName string, sort int, param string) (err error) {
+	// 查询是否包含 id_del 字段
+	hasDeletedAt := checkHasDeletedAt(tx, tableName)
+
+	// 处理排序,如果没有传，就会是在最前面
+	sel := tx.Table(tableName).Where("sort >= ?", sort)
+	if hasDeletedAt {
+		sel.Where("id_del = 0")
+	}
+	if param != "" {
+		sel.Where(param)
+	}
+	err = sel.Update("sort", gorm.Expr("sort + 1")).Error
+	if err != nil {
+		log.Errorf("处理前置排序失败：%v", err)
+	}
+	return
+}
+
+// DealSortAfter
+// @description 处理序号之后
+// @param tx *gorm.DB "已开启的事务对象"
+// @param modelName string "表名"
+// @return error "错误信息"
+func DealSortAfter(tx *gorm.DB, modelName, param string) (err error) {
+	// 保存成功，刷新排序
+	if param != "" {
+		param = " AND " + param
+	}
+
+	// 查询是否包含 id_del 字段
+	hasDeletedAt := checkHasDeletedAt(tx, modelName)
+	if hasDeletedAt {
+		param += " AND id_del = 0"
+	}
+
+	// 如果有更新时间字段，也更新一下值
+	updateParam := ""
+	if checkHasUpdatedAt(tx, modelName) {
+		updateParam = ",updated_at = NOW()"
+	}
+
+	sql := fmt.Sprintf("UPDATE %s a, (SELECT (@i := @i + 1) i, id FROM %s WHERE 1=1 %s order by sort ASC) i, "+
+		"(SELECT @i := 0) ir SET a.sort = i.i %s WHERE a.id = i.id", modelName, modelName, param, updateParam)
+	err = tx.Exec(sql).Error
+	if err != nil {
+		log.Errorf("刷新排序失败: %v", err.Error())
+	}
+	return
+}

model/param/menu/save.go

@@ -0,0 +1,16 @@
+package menu
+
+import "wechat-robot/pkg/types"
+
+// Save
+// @description: 保存菜单入参
+type Save struct {
+	Id       string         `json:"id" form:"id" comment:"菜单Id"`
+	Type     types.MenuType `json:"type" form:"type" comment:"菜单类型(菜单或按钮)" validate:"oneof=MENU BUTTON"`
+	Name     string         `json:"name" form:"name" comment:"页面组件名称"`
+	Path     string         `json:"path" form:"path" comment:"访问路径"`
+	Title    string         `json:"title" form:"title" comment:"菜单标题"`
+	Icon     string         `json:"icon" form:"icon" comment:"菜单图标"`
+	Sort     int            `json:"sort" form:"sort" comment:"排序值(数字越小越靠前)"`
+	ParentId string         `json:"parentId" form:"parentId" comment:"父级菜单ID"`
+}

router/admin/menu.go

@@ -9,5 +9,6 @@ import (
 // @description: 菜单相关
 // @param g
 func menu(g *gin.RouterGroup) {
-	g.GET("/self", menuApi.GetUserMenuTree)
+	g.GET("/self", menuApi.GetUserMenuTree) // 获取当前用户菜单
+	g.POST("", menuApi.Save)                // 保存菜单
 }

service/menu/save.go

@@ -0,0 +1,77 @@
+package menu
+
+import (
+	"fmt"
+	"wechat-robot/internal/database"
+	"wechat-robot/internal/orm"
+	"wechat-robot/model/entity"
+	menuParam "wechat-robot/model/param/menu"
+)
+
+// Save
+// @description: 保存菜单
+// @param param menuParam.Save 菜单数据
+// @return err error 错误信息
+func Save(param menuParam.Save) (err error) {
+	tx := database.Client.Begin()
+	defer func() {
+		if err != nil {
+			tx.Rollback()
+			return
+		}
+		tx.Commit()
+	}()
+
+	// 提前梳理出参数
+	paramSql := "parent_id IS NULL"
+	if param.ParentId != "" {
+		paramSql = fmt.Sprintf("parent_id = '%s'", param.ParentId)
+	}
+
+	if param.Id == "" {
+		// 新增
+		var m entity.Menu
+		// 预处理排序
+		if err = orm.CreateSortBefore(tx, m.TableName(), param.Sort, paramSql); err != nil {
+			return
+		}
+		// 组装参数
+		m.Type = param.Type
+		m.Name = param.Name
+		m.Path = param.Path
+		m.Title = param.Title
+		m.Icon = param.Icon
+		m.Sort = param.Sort
+		if param.ParentId != "" {
+			m.ParentId = &param.ParentId
+		}
+		err = tx.Create(&m).Error
+	} else {
+		// 修改
+		// 预处理排序
+		if err = orm.UpdateSortBefore(tx, entity.Menu{}.TableName(), param.Id, param.Sort, paramSql); err != nil {
+			return
+		}
+		// 组装参数
+		var m = make(map[string]any)
+		m["type"] = param.Type
+		m["name"] = param.Name
+		m["path"] = param.Path
+		m["title"] = param.Title
+		m["icon"] = param.Icon
+		m["sort"] = param.Sort
+		if param.ParentId != "" {
+			m["parent_id"] = &param.ParentId
+		}
+		err = tx.Model(&entity.Menu{}).Where("id = ?", param.Id).Updates(m).Error
+	}
+
+	if err != nil {
+		return
+	}
+
+	// 处理排序
+	err = orm.DealSortAfter(tx, entity.Menu{}.TableName(), paramSql)
+
+	return
+}