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 |   31 +++++++++++++++----------------
 1 files changed, 15 insertions(+), 16 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 28f84b7..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;
 
@@ -145,13 +139,15 @@
 		user.setDelFlag("0");
 		user.setRoleId(partner.getRoleId().longValue());
 		user.setRoleType(2);
-		user.setObjectId(partner.getId());
-		//添加新数据
-		this.save(partner);
-		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,12 +280,12 @@
 	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>().in(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("该合作商已关联充电桩不可删除!");
 		}
@@ -298,7 +294,7 @@
 			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