Pu Zhibing
2024-11-01 a8996df2df6a108be7db1cdb42cee31e12d59245
ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysRoleController.java
@@ -112,6 +112,8 @@
            List<String> collect = sites.stream().map(Site::getName).collect(Collectors.toList());
            record.setSiteNames(collect);
         }
         long count = sysUserRoleService.count(new LambdaQueryWrapper<SysUserRole>().eq(SysUserRole::getRoleId, record.getRoleId()));
         record.setNumber(count);
      }
      return AjaxResult.success(page);
   }
@@ -238,11 +240,13 @@
      roleSiteClient.delRoleSite(dto.getRoleId());
      //添加站点权限
      List<TRoleSite> roleSites = new ArrayList<>();
      for (Integer siteId : dto.getSiteIds()) {
         TRoleSite roleSite = new TRoleSite();
         roleSite.setRoleId(role.getRoleId().intValue());
         roleSite.setSiteId(siteId);
         roleSites.add(roleSite);
      if(null != dto.getSiteIds()){
         for (Integer siteId : dto.getSiteIds()) {
            TRoleSite roleSite = new TRoleSite();
            roleSite.setRoleId(role.getRoleId().intValue());
            roleSite.setSiteId(siteId);
            roleSites.add(roleSite);
         }
      }
      roleSiteClient.addRoleSite(roleSites);
      return AjaxResult.success();