修改角色菜单

2023-07-04 19:09:09 +08:00
parent b393570e86
commit 2944d42281
2 changed files with 22 additions and 6 deletions
@@ -140,7 +140,7 @@ func (r *RoleService) PutRoleStatus(roleId int64, roleStatus int) (bool, error)
 }

 // GetRole 角色详情
-func (r *RoleService) GetRole(roleId int64) (adminRole *model.AdminRole, menuIds []int64, err error) {
+func (r *RoleService) GetRole(roleId int64, isAdmin bool) (adminRole *model.AdminRole, menuIds []int64, err error) {
 	// 获取请求角色数据
 	AdminRoleDao := dao.AdminRoleDao{}
 	adminRole, err = AdminRoleDao.GetAdminRoleFirstById(roleId)
@@ -149,10 +149,18 @@ func (r *RoleService) GetRole(roleId int64) (adminRole *model.AdminRole, menuIds
 	}

 	// 获取角色权限菜单id
-	AdminRoleMenuDao := dao.AdminRoleMenuDao{}
-	menuIds, err = AdminRoleMenuDao.GetAdminRoleMenuIdByRoleId(roleId)
-	if err != nil {
-		return nil, nil, errors.New("角色权限菜单获取失败")
+	if isAdmin {
+		adminMenuDao := dao.AdminMenuDao{}
+		menuIds, err = adminMenuDao.GetAdminMenuId()
+		if err != nil {
+			return nil, nil, errors.New("角色权限菜单获取失败")
+		}
+	} else {
+		AdminRoleMenuDao := dao.AdminRoleMenuDao{}
+		menuIds, err = AdminRoleMenuDao.GetAdminRoleMenuIdByRoleId(roleId)
+		if err != nil {
+			return nil, nil, errors.New("角色权限菜单获取失败")
+		}
 	}

 	return adminRole, menuIds, nil