| | |
| | | */ |
| | | @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(); |
| | | } |
| | | } |
| | | |