From 74b0e0814e37d640596f44ec86d20fa9ecce9ed6 Mon Sep 17 00:00:00 2001
From: puzhibing <393733352@qq.com>
Date: 星期三, 05 二月 2025 10:42:06 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/SiteServiceImpl.java |   36 ++++++++++++++++++++++++------------
 1 files changed, 24 insertions(+), 12 deletions(-)

diff --git a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/SiteServiceImpl.java b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/SiteServiceImpl.java
index 72a8e3e..aa55790 100644
--- a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/SiteServiceImpl.java
+++ b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/SiteServiceImpl.java
@@ -15,12 +15,10 @@
 import com.ruoyi.chargingPile.api.query.SiteQuery;
 import com.ruoyi.chargingPile.api.vo.SiteDetailVO;
 import com.ruoyi.chargingPile.api.vo.SiteVO;
-import com.ruoyi.chargingPile.api.vo.TAccountingStrategyVO;
-import com.ruoyi.chargingPile.domain.SiteMenu;
+import com.ruoyi.chargingPile.api.domain.SiteMenu;
 import com.ruoyi.chargingPile.mapper.SiteMapper;
 import com.ruoyi.chargingPile.service.*;
 import com.ruoyi.common.core.domain.R;
-import com.ruoyi.common.core.utils.ServletUtils;
 import com.ruoyi.common.core.utils.StringUtils;
 import com.ruoyi.common.core.web.domain.AjaxResult;
 import com.ruoyi.common.core.web.page.PageInfo;
@@ -35,8 +33,8 @@
 import com.ruoyi.system.api.feignClient.SysUserClient;
 import com.ruoyi.system.api.feignClient.SysUserRoleClient;
 import com.ruoyi.system.api.model.SysUserRoleVo;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
-import org.springframework.transaction.annotation.Propagation;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.CollectionUtils;
 
@@ -88,6 +86,9 @@
 	@Resource
 	private TAccountingStrategyDetailService accountingStrategyDetailService;
 	
+	@Resource
+	private TPartnerSiteService partnerSiteService;
+	
 	
 	/**
 	 * 获取站点管理列表数据
@@ -99,10 +100,10 @@
 	public PageInfo<GetSiteListDTO> getSiteList(GetSiteList siteList) {
 		Set<Integer> ids = null;
 		//校验合作商权限
-		SysUser sysUser = sysUserClient.getSysUser(SecurityUtils.getUserId()).getData();
+		SysUser sysUser = sysUserClient.getSysUser(tokenService.getLoginUser().getUserid()).getData();
 		Integer roleType = sysUser.getRoleType();
 		Integer objectId = sysUser.getObjectId();
-		R<Integer> admin = sysUserClient.isAdmin(SecurityUtils.getUserId());
+		R<Integer> admin = sysUserClient.isAdmin(tokenService.getLoginUser().getUserid());
 		//合作商
 		if(roleType == 2){
 			ids = partnerService.authSite(objectId, SiteMenu.SITE_LIST);
@@ -110,7 +111,6 @@
 			//非管理员需要根据角色和用户配置查询允许的站点数据
 			//改用roleId=1来判断
 			if(admin.getData()!=1){
-//			if(!SecurityUtils.isAdmin(SecurityUtils.getUserId())){
 				List<Integer> data = userSiteClient.getSiteIds(sysUser.getUserId()).getData();
 				List<SysUserRoleVo> data2 = sysUserRoleClient.getRoleByUserId(sysUser.getUserId()).getData();
 				if(data2.size() > 0){
@@ -134,6 +134,8 @@
 	}
 
 
+
+
 	@Override
 	public PageInfo<GetSiteListDTO> getSiteList1(Long userId) {
 		GetSiteList siteList = new GetSiteList();
@@ -144,7 +146,7 @@
 		SysUser sysUser = sysUserClient.getSysUser(userId).getData();
 		Integer roleType = sysUser.getRoleType();
 		Integer objectId = sysUser.getObjectId();
-		R<Integer> admin = sysUserClient.isAdmin(SecurityUtils.getUserId());
+		R<Integer> admin = sysUserClient.isAdmin(tokenService.getLoginUser().getUserid());
 
 		//合作商
 		if(roleType == 2){
@@ -211,6 +213,16 @@
 		if(null != one && !one.getCode().equals(site.getCode())){
 			return AjaxResult.error("站点编号重复");
 		}
+		
+		Site site1 = this.getById(site.getId());
+		if(!site1.getPartnerId().equals(site.getPartnerId())){
+			List<TPartnerSite> list = partnerSiteService.list(new LambdaQueryWrapper<TPartnerSite>().eq(TPartnerSite::getPartnerId, site1.getPartnerId()).eq(TPartnerSite::getSiteId, site.getId()));
+			for (TPartnerSite partnerSite : list) {
+				partnerSite.setPartnerId(site.getPartnerId());
+				partnerSiteService.updateById(partnerSite);
+			}
+		}
+		
 		this.updateById(site);
 		return AjaxResult.success();
 	}
@@ -334,7 +346,7 @@
 	public List<Site> getSiteListParkLot() {
 		Set<Integer> ids = null;
 		//校验合作商权限
-		SysUser sysUser = sysUserClient.getSysUser(SecurityUtils.getUserId()).getData();
+		SysUser sysUser = sysUserClient.getSysUser(tokenService.getLoginUser().getUserid()).getData();
 		Integer roleType = sysUser.getRoleType();
 		Integer objectId = sysUser.getObjectId();
 		//合作商
@@ -342,7 +354,7 @@
 			ids = partnerService.authSite(objectId, SiteMenu.SITE_LIST);
 		}else{
 			//非管理员需要根据角色和用户配置查询允许的站点数据
-			if(!SecurityUtils.isAdmin(SecurityUtils.getUserId())){
+			if(!SecurityUtils.isAdmin(tokenService.getLoginUser().getUserid())){
 				List<Integer> data = userSiteClient.getSiteIds(sysUser.getUserId()).getData();
 				List<SysUserRoleVo> data2 = sysUserRoleClient.getRoleByUserId(sysUser.getUserId()).getData();
 				List<Integer> data1 = roleSiteClient.getSiteIds(data2.get(0).getRoleId()).getData();
@@ -369,7 +381,7 @@
 	public List<Site> getSiteListGun() {
 		Set<Integer> ids = null;
 		//校验合作商权限
-		SysUser sysUser = sysUserClient.getSysUser(SecurityUtils.getUserId()).getData();
+		SysUser sysUser = sysUserClient.getSysUser(tokenService.getLoginUser().getUserid()).getData();
 		Integer roleType = sysUser.getRoleType();
 		Integer objectId = sysUser.getObjectId();
 		//合作商
@@ -377,7 +389,7 @@
 			ids = partnerService.authSite(objectId, SiteMenu.SITE_LIST);
 		}else{
 			//非管理员需要根据角色和用户配置查询允许的站点数据
-			if(!SecurityUtils.isAdmin(SecurityUtils.getUserId())){
+			if(!SecurityUtils.isAdmin(tokenService.getLoginUser().getUserid())){
 				List<Integer> data = userSiteClient.getSiteIds(sysUser.getUserId()).getData();
 				List<SysUserRoleVo> data2 = sysUserRoleClient.getRoleByUserId(sysUser.getUserId()).getData();
 				List<Integer> data1 = roleSiteClient.getSiteIds(data2.get(0).getRoleId()).getData();

--
Gitblit v1.7.1