lc_veterans/app/system/service/roleMenuService.go

package service

import "iot_manager_service/app/system/dao"

// 角色菜单管理服务
var RoleMenuService = new(roleMenuService)

type roleMenuService struct{}

func (s *roleMenuService) UpdateRoleMenus(roleIds, menuIds []int64) error {
	roleMenu := &dao.RoleMenu{}
	err := roleMenu.DeleteRoleIds(roleIds)
	if err != nil {
		return err
	}

	var roleMenus []dao.RoleMenu
	for _, roleId := range roleIds {
		for _, menuId := range menuIds {
			roleMenus = append(roleMenus, dao.RoleMenu{
				MenuId: menuId,
				RoleId: roleId,
			})
		}
	}
	return roleMenu.BatchSave(roleMenus)
}

func (s *roleMenuService) GetMenuByRole(roleId int64) []dao.RoleMenu {
	roleMenu := &dao.RoleMenu{RoleId: roleId}
	menus, err := roleMenu.GetByRole()
	if err != nil {
		return nil
	}
	return menus
}