From 67d06a785d5fc4524639ace27651c01ff05a4b58 Mon Sep 17 00:00:00 2001
From: liujie <1793218484@qq.com>
Date: 星期四, 29 五月 2025 22:31:33 +0800
Subject: [PATCH] 修改bug
---
ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TUserSiteController.java | 51 +++++++++++++++++++++++++++++++++++++++++++++++++--
1 files changed, 49 insertions(+), 2 deletions(-)
diff --git a/ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TUserSiteController.java b/ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TUserSiteController.java
index 4800cb2..8b5327c 100644
--- a/ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TUserSiteController.java
+++ b/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();
+ }
}
--
Gitblit v1.7.1