From f47d695f19bca1e58ddedcc10f8d3dff7d7f8f1d Mon Sep 17 00:00:00 2001
From: zhibing.pu <393733352@qq.com>
Date: 星期三, 21 八月 2024 11:39:43 +0800
Subject: [PATCH] Merge branch 'master' of http://120.76.84.145:10101/gitblit/r/java/mx_charging_pile

---
 ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TRoleSiteController.java |   49 +++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 47 insertions(+), 2 deletions(-)

diff --git a/ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TRoleSiteController.java b/ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TRoleSiteController.java
index 0fa48ec..089747c 100644
--- a/ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TRoleSiteController.java
+++ b/ruoyi-service/ruoyi-other/src/main/java/com/ruoyi/other/controller/TRoleSiteController.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.TRoleSite;
+import com.ruoyi.other.service.TRoleSiteService;
+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,44 @@
 @RequestMapping("/t-role-site")
 public class TRoleSiteController {
 
+	@Resource
+	private TRoleSiteService roleSiteService;
+	
+	/**
+	 * 查询角色站点数据
+	 * @param roleId
+	 * @return
+	 */
+	@ResponseBody
+	@PostMapping("/getSiteIds/{roleId}")
+	public R<List<Integer>> getSiteIds(@PathVariable Long roleId){
+		List<TRoleSite> list = roleSiteService.list(new LambdaQueryWrapper<TRoleSite>().eq(TRoleSite::getRoleId, roleId));
+		List<Integer> collect = list.stream().map(TRoleSite::getSiteId).collect(Collectors.toList());
+		return R.ok(collect);
+	}
+	
+	/**
+	 * 添加角色站点数据
+	 * @param roleSites
+	 * @return
+	 */
+	@ResponseBody
+	@PostMapping("/addRoleSite")
+	public R addRoleSite(@RequestBody List<TRoleSite> roleSites){
+		roleSiteService.saveBatch(roleSites);
+		return R.ok();
+	}
+	
+	/**
+	 * 删除角色站点数据
+	 * @param roleId
+	 * @return
+	 */
+	@ResponseBody
+	@DeleteMapping("/delRoleSite/{roleId}")
+	public R delRoleSite(@PathVariable("roleId") Long roleId){
+		roleSiteService.remove(new LambdaQueryWrapper<TRoleSite>().eq(TRoleSite::getRoleId, roleId));
+		return R.ok();
+	}
 }
 

--
Gitblit v1.7.1