sys_base_menu.go 2.8 KB
Newer Older
1 2 3 4 5 6
package service

import (
	"errors"
	"gin-vue-admin/global"
	"gin-vue-admin/model"
Mr.奇淼('s avatar
Mr.奇淼( 已提交
7
	"gorm.io/gorm"
8 9
)

m0_50812349's avatar
m0_50812349 已提交
10 11 12 13 14
//@author: [piexlmax](https://github.com/piexlmax)
//@function: DeleteBaseMenu
//@description: 删除基础路由
//@param: id float64
//@return: err error
Mr.奇淼('s avatar
Mr.奇淼( 已提交
15 16

func DeleteBaseMenu(id float64) (err error) {
Mr.奇淼('s avatar
Mr.奇淼( 已提交
17
	err = global.GVA_DB.Preload("Parameters").Where("parent_id = ?", id).First(&model.SysBaseMenu{}).Error
18 19 20
	if err != nil {
		var menu model.SysBaseMenu
		db := global.GVA_DB.Preload("SysAuthoritys").Where("id = ?", id).First(&menu).Delete(&menu)
Mr.奇淼('s avatar
Mr.奇淼( 已提交
21
		err = global.GVA_DB.Delete(&model.SysBaseMenuParameter{}, "sys_base_menu_id = ?", id).Error
22
		if len(menu.SysAuthoritys) > 0 {
Mr.奇淼('s avatar
Mr.奇淼( 已提交
23
			err = global.GVA_DB.Model(&menu).Association("SysAuthoritys").Delete(&menu.SysAuthoritys)
24 25 26 27 28 29 30 31 32
		} else {
			err = db.Error
		}
	} else {
		return errors.New("此菜单存在子菜单不可删除")
	}
	return err
}

m0_50812349's avatar
m0_50812349 已提交
33 34 35 36
//@author: [piexlmax](https://github.com/piexlmax)
//@function: UpdateBaseMenu
//@description: 更新路由
//@param: menu model.SysBaseMenu
何秀钢 已提交
37
//@return: err error
Mr.奇淼('s avatar
Mr.奇淼( 已提交
38

S
songzhibin97 已提交
39
func UpdateBaseMenu(menu model.SysBaseMenu) (err error) {
40
	var oldMenu model.SysBaseMenu
41
	upDateMap := make(map[string]interface{})
42
	upDateMap["keep_alive"] = menu.KeepAlive
Mr.奇淼('s avatar
Mr.奇淼( 已提交
43
	upDateMap["close_tab"] = menu.CloseTab
44
	upDateMap["default_menu"] = menu.DefaultMenu
45 46 47 48 49 50 51 52
	upDateMap["parent_id"] = menu.ParentId
	upDateMap["path"] = menu.Path
	upDateMap["name"] = menu.Name
	upDateMap["hidden"] = menu.Hidden
	upDateMap["component"] = menu.Component
	upDateMap["title"] = menu.Title
	upDateMap["icon"] = menu.Icon
	upDateMap["sort"] = menu.Sort
Mr.奇淼('s avatar
Mr.奇淼( 已提交
53

Mr.奇淼('s avatar
Mr.奇淼( 已提交
54 55 56 57 58 59 60
	err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		db := tx.Where("id = ?", menu.ID).Find(&oldMenu)
		if oldMenu.Name != menu.Name {
			if !errors.Is(tx.Where("id <> ? AND name = ?", menu.ID, menu.Name).First(&model.SysBaseMenu{}).Error, gorm.ErrRecordNotFound) {
				global.GVA_LOG.Debug("存在相同name修改失败")
				return errors.New("存在相同name修改失败")
			}
61
		}
62 63 64 65
		txErr := tx.Unscoped().Delete(&model.SysBaseMenuParameter{}, "sys_base_menu_id = ?", menu.ID).Error
		if txErr != nil {
			global.GVA_LOG.Debug(txErr.Error())
			return txErr
Mr.奇淼('s avatar
Mr.奇淼( 已提交
66 67 68 69 70
		}
		if len(menu.Parameters) > 0 {
			for k, _ := range menu.Parameters {
				menu.Parameters[k].SysBaseMenuID = menu.ID
			}
71 72 73 74
			txErr = tx.Create(&menu.Parameters).Error
			if txErr != nil {
				global.GVA_LOG.Debug(txErr.Error())
				return txErr
Mr.奇淼('s avatar
Mr.奇淼( 已提交
75 76 77
			}
		}

78 79 80 81
		txErr = db.Updates(upDateMap).Error
		if txErr != nil {
			global.GVA_LOG.Debug(txErr.Error())
			return txErr
Mr.奇淼('s avatar
Mr.奇淼( 已提交
82 83 84
		}
		return nil
	})
85 86 87
	return err
}

m0_50812349's avatar
m0_50812349 已提交
88 89 90 91 92
//@author: [piexlmax](https://github.com/piexlmax)
//@function: GetBaseMenuById
//@description: 返回当前选中menu
//@param: id float64
//@return: err error, menu model.SysBaseMenu
Mr.奇淼('s avatar
Mr.奇淼( 已提交
93

94
func GetBaseMenuById(id float64) (err error, menu model.SysBaseMenu) {
Mr.奇淼('s avatar
Mr.奇淼( 已提交
95
	err = global.GVA_DB.Preload("Parameters").Where("id = ?", id).First(&menu).Error
96 97
	return
}