cas_tt_cloud_backend/app/api/admin/controller/role/menu.go

package role

import (
	menu2 "SciencesServer/app/api/admin/controller/menu"
	"SciencesServer/app/api/admin/model"
	model2 "SciencesServer/app/common/model"
	"SciencesServer/app/session"
	"SciencesServer/serve/orm"
	"gorm.io/gorm"
)

type Menu struct{ *session.Admin }

type MenuHandle func(session *session.Admin) *Menu

// Index 菜单列表
func (c *Menu) Index(roleID uint64) ([]*menu2.TreeChecked, error) {
	mSysMenu := model.NewSysMenu()
	return menu2.MenuForRoleChecked(mSysMenu, c.TenantID, roleID)
}

// Bind 绑定菜单
func (c *Menu) Bind(roleID uint64, menuIDs []uint64) error {
	if len(menuIDs) > 0 {

	}
	mSysRoleMenu := model.NewSysRoleMenu()

	return orm.GetDB().Transaction(func(tx *gorm.DB) error {
		err := model2.DeleteWhere(mSysRoleMenu.SysRoleMenu, []*model2.ModelWhere{model2.NewWhere("role_id", roleID)}, tx)

		if err != nil {
			return err
		}
		if len(menuIDs) <= 0 {
			return nil
		}
		menus := make([]*model2.SysRoleMenu, 0)

		mark := make(map[uint64]uint64, 0)

		for _, v := range menuIDs {
			if _, has := mark[v]; has {
				continue
			}
			menus = append(menus, &model2.SysRoleMenu{RoleID: roleID, MenuID: v})
			mark[v] = v
		}
		if err = model2.Creates(mSysRoleMenu.SysRoleMenu, menus, tx); err != nil {
			return err
		}
		return nil
	})
}

func NewMenu() MenuHandle {
	return func(session *session.Admin) *Menu {
		return &Menu{Admin: session}
	}
}