go-wxhelper/service/menu/save.go

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
}