From c05a14bba4b86d83918d7cdc01635785bf5694c2 Mon Sep 17 00:00:00 2001
From: Pu Zhibing <393733352@qq.com>
Date: 星期二, 03 六月 2025 00:26:08 +0800
Subject: [PATCH] 拆分mongodb查询服务

---
 ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/PartnerServiceImpl.java |   33 ++++++++++++++++-----------------
 1 files changed, 16 insertions(+), 17 deletions(-)

diff --git a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/PartnerServiceImpl.java b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/PartnerServiceImpl.java
index 2ad2b83..f56c4b4 100644
--- a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/PartnerServiceImpl.java
+++ b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/PartnerServiceImpl.java
@@ -7,7 +7,7 @@
 import com.ruoyi.chargingPile.api.model.*;
 import com.ruoyi.chargingPile.api.query.GetPartnerList;
 import com.ruoyi.chargingPile.api.query.SetPermissionConfiguration;
-import com.ruoyi.chargingPile.domain.SiteMenu;
+import com.ruoyi.chargingPile.api.domain.SiteMenu;
 import com.ruoyi.chargingPile.dto.ResetPassword;
 import com.ruoyi.chargingPile.mapper.PartnerMapper;
 import com.ruoyi.chargingPile.service.*;
@@ -15,7 +15,6 @@
 import com.ruoyi.common.core.utils.StringUtils;
 import com.ruoyi.common.core.web.domain.AjaxResult;
 import com.ruoyi.common.core.web.page.PageInfo;
-import com.ruoyi.common.security.utils.SecurityUtils;
 import com.ruoyi.other.api.feignClient.RoleSiteClient;
 import com.ruoyi.other.api.feignClient.UserSiteClient;
 import com.ruoyi.system.api.domain.SysRole;
@@ -25,12 +24,7 @@
 import com.ruoyi.system.api.feignClient.SysUserRoleClient;
 import com.ruoyi.system.api.model.GetSysRoleByIds;
 import com.ruoyi.system.api.model.SysUserRoleVo;
-import io.seata.core.context.RootContext;
-import io.seata.core.exception.TransactionException;
 import io.seata.spring.annotation.GlobalTransactional;
-import io.seata.tm.api.GlobalTransaction;
-import io.seata.tm.api.GlobalTransactionContext;
-import org.springframework.http.HttpStatus;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -94,10 +88,10 @@
 		List<SysRole> data = r.getData();
 		for (PartnerListDTO partnerListDTO : list) {
 			Integer roleId = partnerListDTO.getRoleId();
-			if(null != roleId){
+			if(null == roleId){
 				continue;
 			}
-			SysRole sysRole = data.stream().filter(s -> roleId.equals(s.getRoleId())).findFirst().get();
+			SysRole sysRole = data.stream().filter(s -> roleId.compareTo(s.getRoleId().intValue()) == 0).findFirst().get();
 			if(null != sysRole){
 				partnerListDTO.setRoleName(sysRole.getRoleName());
 			}
@@ -145,13 +139,15 @@
 		user.setDelFlag("0");
 		user.setRoleId(partner.getRoleId().longValue());
 		user.setRoleType(2);
-		user.setObjectId(partner.getId());
-		R r = sysUserClient.addSysUser(user);
+		R<Long> r = sysUserClient.addSysUser(user);
 		if(200 != r.getCode()){
 			throw new RuntimeException(r.getMsg());
 		}
 		//添加新数据
 		this.save(partner);
+		user = sysUserClient.getSysUser(r.getData()).getData();
+		user.setObjectId(partner.getId());
+		sysUserClient.updateUser(user);
 		return AjaxResult.success();
 	}
 	
@@ -284,21 +280,21 @@
 	public AjaxResult delPartner(List<Integer> ids) {
 		//查询有无关联数据
 		//站点
-		long count = siteService.count(new LambdaQueryWrapper<Site>().in(Site::getPartnerId, Arrays.asList(ids)).eq(Site::getDelFlag, 0));
+		long count = siteService.count(new LambdaQueryWrapper<Site>().in(Site::getPartnerId, ids).eq(Site::getDelFlag, 0));
 		if(count > 0){
-			return AjaxResult.error("当前合作商有关联站点,删除失败!");
+			return AjaxResult.error("该合作商已关联站点不可删除");
 		}
 		//充电桩
-		long count1 = chargingPileService.count(new LambdaQueryWrapper<TChargingPile>().eq(TChargingPile::getPartnerId, Arrays.asList(ids)).eq(TChargingPile::getDelFlag, 0));
+		long count1 = chargingPileService.count(new LambdaQueryWrapper<TChargingPile>().in(TChargingPile::getPartnerId, ids).eq(TChargingPile::getDelFlag, 0));
 		if(count1 > 0){
-			return AjaxResult.error("当前合作商有关联充电桩,删除失败!");
+			return AjaxResult.error("该合作商已关联充电桩不可删除!");
 		}
 		for (Integer id : ids) {
 			Partner partner = this.getById(id);
 			if(null == partner){
 				return AjaxResult.error("删除失败");
 			}
-			this.removeById(partner);
+			this.removeById(id);
 			SysUser user = sysUserClient.queryUserByUserName(partner.getAccount()).getData();
 			user.setDelFlag("2");
 			sysUserClient.updateUser(user);
@@ -338,6 +334,9 @@
 	@Override
 	public GetPermissionConfigurationDTO getPermissionConfiguration(Integer siteId, Integer partnerId) {
 		Partner partner = this.getById(partnerId);
+		if(null == partner){
+			throw new RuntimeException("该站点合作商无效");
+		}
 		GetPermissionConfigurationDTO dto = new GetPermissionConfigurationDTO();
 		dto.setPermissionRemarks(partner.getPermissionRemarks());
 		List<TPartnerSite> list = partnerSiteService.list(new LambdaQueryWrapper<TPartnerSite>()
@@ -370,7 +369,7 @@
 				child.put("selected", collect.contains(siteMenu.getMenuId()));
 				list2.add(child);
 			}
-			menu.put("selected", list2.size() > 0 ? true : false);
+			menu.put("selected", collect.contains(tSiteMenu.getMenuId()));
 			menu.put("child", list2);
 			list1.add(menu);
 		}

--
Gitblit v1.7.1