package com.ruoyi.other.controller;
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.ruoyi.common.core.domain.R;
|
import com.ruoyi.other.api.domain.TUserSite;
|
import com.ruoyi.other.service.TUserSiteService;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.annotation.Resource;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 前端控制器
|
* </p>
|
*
|
* @author 无关风月
|
* @since 2024-08-06
|
*/
|
@RestController
|
@RequestMapping("/t-user-site")
|
public class TUserSiteController {
|
|
@Resource
|
private TUserSiteService userSiteService;
|
|
|
/**
|
* 根据用户id获取可访问的站点数据
|
* @param userId 用户id
|
* @return
|
*/
|
@ResponseBody
|
@GetMapping("/getSiteIds")
|
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();
|
}
|
}
|
