zhibing.pu
2024-08-10 1916f55359bb6c8f34548c26767569b000decbfc
ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TUserSiteController.java
@@ -34,12 +34,35 @@
    */
   @ResponseBody
   @GetMapping("/getSiteIds")
   public R<List<Integer>> getSiteIds(@RequestParam("userId") Long userId){
   public R<List<Integer>> getSiteIds(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();
   }
}