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 }