From 8b7258c9427882bb1798f1502eaa35184c6e374e Mon Sep 17 00:00:00 2001
From: linwe <872216996@qq.com>
Date: Fri, 09 Aug 2024 14:29:18 +0800
Subject: [PATCH] 短信指定楼栋发送

---
 src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java |  142 ++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 121 insertions(+), 21 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 5a320a1..62d13ee 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
@@ -1,28 +1,30 @@
-/*
- *      Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
- *
- *  Redistribution and use in source and binary forms, with or without
- *  modification, are permitted provided that the following conditions are met:
- *
- *  Redistributions of source code must retain the above copyright notice,
- *  this list of conditions and the following disclaimer.
- *  Redistributions in binary form must reproduce the above copyright
- *  notice, this list of conditions and the following disclaimer in the
- *  documentation and/or other materials provided with the distribution.
- *  Neither the name of the dreamlu.net developer nor the names of its
- *  contributors may be used to endorse or promote products derived from
- *  this software without specific prior written permission.
- *  Author: Chill 庄骞 (smallchill@163.com)
- */
 package org.springblade.modules.property.service.impl;
 
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.flowable.idm.engine.impl.persistence.entity.UserEntity;
+import org.springblade.common.exception.CustomException;
+import org.springblade.common.utils.RoleUtil;
+import org.springblade.common.utils.SpringUtils;
+import org.springblade.core.secure.utils.AuthUtil;
+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.vo.PropertyCompanyDistrictVO;
+import org.springblade.modules.property.entity.PropertyCompanyEntity;
+import org.springblade.modules.property.entity.PropertyDistrictUserEntity;
 import org.springblade.modules.property.mapper.PropertyCompanyDistrictMapper;
 import org.springblade.modules.property.service.IPropertyCompanyDistrictService;
-import org.springblade.core.mp.base.BaseServiceImpl;
+import org.springblade.modules.property.service.IPropertyCompanyService;
+import org.springblade.modules.property.service.IPropertyDistrictUserService;
+import org.springblade.modules.property.vo.PropertyCompanyDistrictVO;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
-import com.baomidou.mybatisplus.core.metadata.IPage;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * 物业派驻小区表 服务实现类
@@ -31,12 +33,110 @@
  * @since 2023-11-23
  */
 @Service
-public class PropertyCompanyDistrictServiceImpl extends BaseServiceImpl<PropertyCompanyDistrictMapper, PropertyCompanyDistrictEntity> implements IPropertyCompanyDistrictService {
+public class PropertyCompanyDistrictServiceImpl extends ServiceImpl<PropertyCompanyDistrictMapper, PropertyCompanyDistrictEntity> implements IPropertyCompanyDistrictService {
 
+	@Autowired
+	private IPropertyDistrictUserService propertyDistrictUserService;
+
+	/**
+	 * 自定义分页查询
+	 *
+	 * @param page
+	 * @param propertyCompanyDistrict
+	 * @return
+	 */
 	@Override
 	public IPage<PropertyCompanyDistrictVO> selectPropertyCompanyDistrictPage(IPage<PropertyCompanyDistrictVO> page, PropertyCompanyDistrictVO propertyCompanyDistrict) {
-		return page.setRecords(baseMapper.selectPropertyCompanyDistrictPage(page, propertyCompanyDistrict));
+		// 物业 查询用户管理的小区
+		String userRole = AuthUtil.getUserRole();
+		if (RoleUtil.isProperty(userRole)) {
+			propertyCompanyDistrict.setDeptId(AuthUtil.getDeptId());
+			// 查询小区id
+			IPropertyDistrictUserService propertyDistrictUserService = SpringUtils.getBean(IPropertyDistrictUserService.class);
+			List<String> districtIds = propertyDistrictUserService.selectPropertyDistrictByUserId(AuthUtil.getUserId());
+			// 通过用户机构查询用户的物业公司
+			IPropertyCompanyService bean = SpringUtil.getBean(IPropertyCompanyService.class);
+			PropertyCompanyEntity one = bean.getOne(Wrappers.<PropertyCompanyEntity>lambdaQuery().eq(PropertyCompanyEntity::getDeptId, AuthUtil.getDeptId()).last("limit 1"));
+			if (one != null) {
+				IPropertyCompanyDistrictService bean2 = SpringUtils.getBean(IPropertyCompanyDistrictService.class);
+				// 通过物业公司,查询小区
+				List<PropertyCompanyDistrictEntity> list = bean2.list(Wrappers.<PropertyCompanyDistrictEntity>lambdaQuery()
+					.eq(PropertyCompanyDistrictEntity::getPropertyCompanyId, one.getId()));
+				if (list.size() > 0) {
+					List<String> collect = list.stream().map(i -> i.getDistrictId()).collect(Collectors.toList());
+					districtIds.addAll(collect);
+				}
+			}
+			propertyCompanyDistrict.setDistrictIds(districtIds);
+			if (districtIds.size() == 0) {
+				return page.setRecords(new ArrayList<>());
+			}
+		}
+		// 街道
+		List<PropertyCompanyDistrictVO> propertyCompanyDistrictVOS = baseMapper.selectPropertyCompanyDistrictPage(page, propertyCompanyDistrict);
+		return page.setRecords(propertyCompanyDistrictVOS);
 	}
 
+	/**
+	 * 物业派驻小区表 自定义新增或修改
+	 *
+	 * @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 {
+			long count = count(Wrappers.<PropertyCompanyDistrictEntity>lambdaQuery()
+				.eq(PropertyCompanyDistrictEntity::getDistrictId, propertyCompanyDistrict.getDistrictId())
+				.eq(PropertyCompanyDistrictEntity::getPropertyCompanyId, propertyCompanyDistrict.getPropertyCompanyId())
+				.eq(PropertyCompanyDistrictEntity::getIsDeleted, 0));
+			if (count > 0) {
+				throw new CustomException("您已有该小区的合同,请勿重复添加!");
+			}
+			// bean.updateById(user);
+			return save(propertyCompanyDistrict) && submitPropertyDistrictUser(propertyCompanyDistrict);
+		}
+	}
 
+	/**
+	 * 关联关系维护
+	 *
+	 * @param propertyCompanyDistrict
+	 * @return
+	 */
+	private boolean submitPropertyDistrictUser(PropertyCompanyDistrictEntity propertyCompanyDistrict) {
+		List<Long> userIdList = Func.toLongList(propertyCompanyDistrict.getUserId());
+		List<Long> arrayList = new ArrayList<Long>(userIdList);
+		// 把项目经理也加到物业用户关联表
+		arrayList.add(Long.valueOf(propertyCompanyDistrict.getPrincipal()));
+		List<PropertyDistrictUserEntity> propertyDistrictUserEntityList = new ArrayList<>();
+		arrayList.forEach(userId -> {
+			PropertyDistrictUserEntity propertyDistrictUserEntity = new PropertyDistrictUserEntity();
+			propertyDistrictUserEntity.setUserId(userId);
+			propertyDistrictUserEntity.setPropertyCompanyDistrictId(propertyCompanyDistrict.getId());
+			propertyDistrictUserEntityList.add(propertyDistrictUserEntity);
+		});
+		// 先删除
+		propertyDistrictUserService.remove(Wrappers.<PropertyDistrictUserEntity>update().lambda().eq(PropertyDistrictUserEntity::getPropertyCompanyDistrictId, propertyCompanyDistrict.getId()));
+		// 再新增
+		return propertyDistrictUserService.saveBatch(propertyDistrictUserEntityList);
+	}
+
+	@Override
+	public List<UserEntity> getDistictUserByCode(String houseCode) {
+		return baseMapper.getDistictUserByCode(houseCode);
+	}
 }

--
Gitblit v1.9.3