From 852cc7a233ea2ff510f1e4ab6e26618680c981f5 Mon Sep 17 00:00:00 2001
From: linwei <872216696@qq.com>
Date: Mon, 22 Jan 2024 15:22:34 +0800
Subject: [PATCH] 合同管理优化

---
 src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java |   20 +++++++++++++++++++-
 1 files changed, 19 insertions(+), 1 deletions(-)

diff --git a/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java b/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java
index 8fce3ad..c61228f 100644
--- a/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java
+++ b/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java
@@ -2,15 +2,19 @@
 
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.apache.commons.lang3.StringUtils;
 import org.flowable.idm.engine.impl.persistence.entity.UserEntity;
 import org.springblade.core.tool.utils.Func;
+import org.springblade.core.tool.utils.SpringUtil;
 import org.springblade.modules.property.entity.PropertyCompanyDistrictEntity;
 import org.springblade.modules.property.entity.PropertyDistrictUserEntity;
 import org.springblade.modules.property.service.IPropertyDistrictUserService;
 import org.springblade.modules.property.vo.PropertyCompanyDistrictVO;
 import org.springblade.modules.property.mapper.PropertyCompanyDistrictMapper;
 import org.springblade.modules.property.service.IPropertyCompanyDistrictService;
+import org.springblade.modules.system.entity.User;
 import org.springblade.modules.system.entity.UserDept;
+import org.springblade.modules.system.service.IUserService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import com.baomidou.mybatisplus.core.metadata.IPage;
@@ -33,6 +37,7 @@
 
 	/**
 	 * 自定义分页查询
+	 *
 	 * @param page
 	 * @param propertyCompanyDistrict
 	 * @return
@@ -44,21 +49,34 @@
 
 	/**
 	 * 物业派驻小区表 自定义新增或修改
+	 *
 	 * @param propertyCompanyDistrict
 	 * @return
 	 */
 	@Override
 	@Transactional(rollbackFor = Exception.class)
 	public boolean saveOrUpdatePropertyCompanyDistrict(PropertyCompanyDistrictEntity propertyCompanyDistrict) {
+		IUserService bean = SpringUtil.getBean(IUserService.class);
+		User user = bean.getOne(Wrappers.<User>lambdaQuery().eq(User::getId, propertyCompanyDistrict.getPrincipal()));
+		if (StringUtils.isNotBlank(user.getRoleId())) {
+			if (!user.getRoleId().contains("1747504028253229058")) {
+				user.setRoleId(user.getRoleId() + ",1747504028253229058");
+			}
+		} else {
+			user.setRoleId("1747504028253229058");
+		}
 		if (null != propertyCompanyDistrict.getId()) {
+			bean.updateById(user);
 			return updateById(propertyCompanyDistrict) && submitPropertyDistrictUser(propertyCompanyDistrict);
-		}else {
+		} else {
+			bean.updateById(user);
 			return save(propertyCompanyDistrict) && submitPropertyDistrictUser(propertyCompanyDistrict);
 		}
 	}
 
 	/**
 	 * 关联关系维护
+	 *
 	 * @param propertyCompanyDistrict
 	 * @return
 	 */

--
Gitblit v1.9.3