Merge remote-tracking branch 'origin/master'

zhibing.pu

2024-08-22 490db2e9cc769bd0aa3571fad345c7a01bc01bdd

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

@@ -1,9 +1,15 @@
package com.ruoyi.other.controller;


import org.springframework.web.bind.annotation.RequestMapping;
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 org.springframework.web.bind.annotation.RestController;
import javax.annotation.Resource;
import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
@@ -17,5 +23,46 @@
@RequestMapping("/t-user-site")
public class TUserSiteController {

   @Resource
   private TUserSiteService userSiteService;
	
	
   /**
    * 根据用户id获取可访问的站点数据
    * @param userId 用户id
    * @return
    */
   @ResponseBody
   @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();
   }
}