zhibing.pu
2024-08-21 f47d695f19bca1e58ddedcc10f8d3dff7d7f8f1d
ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TRoleSiteController.java
@@ -26,15 +26,41 @@
   @Resource
   private TRoleSiteService roleSiteService;
   
   /**
    * 查询角色站点数据
    * @param roleId
    * @return
    */
   @ResponseBody
   @GetMapping("/getSiteIds")
   public R<List<Integer>> getSiteIds(@RequestParam("roleId") Long roleId){
   @PostMapping("/getSiteIds/{roleId}")
   public R<List<Integer>> getSiteIds(@PathVariable Long roleId){
      List<TRoleSite> list = roleSiteService.list(new LambdaQueryWrapper<TRoleSite>().eq(TRoleSite::getRoleId, roleId));
      List<Integer> collect = list.stream().map(TRoleSite::getSiteId).collect(Collectors.toList());
      return R.ok(collect);
   }
   
   /**
    * 添加角色站点数据
    * @param roleSites
    * @return
    */
   @ResponseBody
   @PostMapping("/addRoleSite")
   public R addRoleSite(@RequestBody List<TRoleSite> roleSites){
      roleSiteService.saveBatch(roleSites);
      return R.ok();
   }
   
   /**
    * 删除角色站点数据
    * @param roleId
    * @return
    */
   @ResponseBody
   @DeleteMapping("/delRoleSite/{roleId}")
   public R delRoleSite(@PathVariable("roleId") Long roleId){
      roleSiteService.remove(new LambdaQueryWrapper<TRoleSite>().eq(TRoleSite::getRoleId, roleId));
      return R.ok();
   }
}