代码合并

puzhibing

2024-08-22 bdee6ad7dd52e083cbf6bca7fc1c7d3ec270f2a4

 ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TUserSiteController.java

@@ -33,13 +33,36 @@
    * @return
    */
   @ResponseBody
   @GetMapping("/getSiteIds")
   public R<List<Integer>> getSiteIds(@RequestParam("userId") Long userId){
   @PostMapping("/getSiteIds/{userId}")
   public R<List<Integer>> getSiteIds(@PathVariable Long userId){
      List<TUserSite> list = userSiteService.list(new LambdaQueryWrapper<TUserSite>().eq(TUserSite::getUserId, userId));
      List<Integer> collect = list.stream().map(TUserSite::getSiteId).collect(Collectors.toList());
      return R.ok(collect);
   }
   
   
   /**
    * 添加用户站点数据
    * @param userSite
    * @return
    */
   @ResponseBody
   @PostMapping("/addUserSite")
   public R addUserSite(@RequestBody List<TUserSite> userSite){
      userSiteService.saveBatch(userSite);
      return R.ok();
   }
	
   /**
    * 删除用户对应的站点数据
    * @param userId
    * @return
    */
   @ResponseBody
   @DeleteMapping("/delUserSite/{userId}")
   public R delUserSite(@PathVariable Long userId){
      userSiteService.remove(new LambdaQueryWrapper<TUserSite>().eq(TUserSite::getUserId, userId));
      return R.ok();
   }
}