From 26cd0115150771d66a6cfca9c03820a734b69c9f Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期三, 11 十二月 2024 19:19:38 +0800
Subject: [PATCH] 合作商权限

---
 ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/SiteController.java |   21 +++++++++++++++------
 1 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/SiteController.java b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/SiteController.java
index f45575b..2677245 100644
--- a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/SiteController.java
+++ b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/SiteController.java
@@ -13,6 +13,7 @@
 import com.ruoyi.account.api.vo.GetAppUserVipDetail;
 import com.ruoyi.chargingPile.api.dto.GetSiteListDTO;
 import com.ruoyi.chargingPile.api.feignClient.ParkingLotClient;
+import com.ruoyi.chargingPile.api.feignClient.SiteClient;
 import com.ruoyi.chargingPile.api.model.*;
 import com.ruoyi.chargingPile.api.query.GetSiteList;
 import com.ruoyi.chargingPile.api.query.SiteDetailQuery;
@@ -325,15 +326,22 @@
 		return R.ok(divide);
 	}
 
-
+	@Resource
+    private SiteClient siteClient;
 
 
 	@PostMapping("/map/getPercentByprovinceCode")
 	@ApiOperation(value = "获取该充电站的利用率", tags = {"管理后台-数据分析-电站分析"})
 	public R getPercentByCityCode(@RequestBody ChargingPercentProvinceDto chargingPercentProvinceDto){
 
+		Long userid = tokenService.getLoginUser().getUserid();
+		List<Integer> siteIds = new ArrayList<>();
 
-
+		List<GetSiteListDTO> data1 = siteClient.getSiteListByUserId(userid).getData();
+		for (GetSiteListDTO datum : data1) {
+			siteIds.add(datum.getId());
+		}
+		chargingPercentProvinceDto.setSiteIds(siteIds);
 
 
 		GetSiteList siteList = new GetSiteList();
@@ -411,8 +419,9 @@
 
 	@ApiOperation(value = "获取站点列表", tags = {"小程序-站点管理-地图导航站点"})
 	@PostMapping("/getMapSiteList")
-	public AjaxResult<List<SiteVO>> getMapSiteList(@Validated @RequestBody SiteQuery query){
-		return AjaxResult.success(siteService.getMapSiteList(query));
+	public AjaxResult<List<SiteVO>> getMapSiteList(@Validated @RequestBody SiteQuery query, HttpServletRequest request){
+		int isLogin = StringUtils.isEmpty(SecurityUtils.getToken(request)) ? 0 : 1;
+		return AjaxResult.success(siteService.getMapSiteList(query,isLogin));
 	}
 	@ApiOperation(value = "获取站点详细信息", tags = {"小程序-站点管理-站点详情"})
 	@PostMapping("/getDetailById")
@@ -508,7 +517,7 @@
 	@ApiOperation(value = "获取所有站点数据", tags = {"管理后台-充电桩信息", "管理后台-系统用户管理", "管理后台-角色管理", "管理后台-系统通知", "管理后台-充电评价"})
 	public AjaxResult<List<Site>> getAllSiteList(){
 		//校验当前账户站点权限
-		SysUser sysUser = sysUserClient.getSysUser(SecurityUtils.getUserId()).getData();
+		SysUser sysUser = sysUserClient.getSysUser(tokenService.getLoginUser().getUserid()).getData();
 		Integer roleType = sysUser.getRoleType();
 		Integer objectId = sysUser.getObjectId();
 		LambdaQueryWrapper<Site> wrapper = new LambdaQueryWrapper<Site>().eq(Site::getDelFlag, 0);
@@ -517,7 +526,7 @@
 			wrapper.in(Site::getId, list);
 		}else{
 			//非管理员需要根据角色和用户配置查询允许的站点数据
-			if(!SecurityUtils.isAdmin(SecurityUtils.getUserId())){
+			if(!SecurityUtils.isAdmin(tokenService.getLoginUser().getUserid())){
 				List<Integer> data = userSiteClient.getSiteIds(sysUser.getUserId()).getData();
 				List<SysUserRoleVo> data2 = userRoleClient.getRoleByUserId(sysUser.getUserId()).getData();
 				List<Integer> data1 = roleSiteClient.getSiteIds(data2.get(0).getRoleId()).getData();

--
Gitblit v1.7.1