linwe
2024-05-29 c10d6358b9f014375a13821465bc978d0c0da22e
src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java
@@ -3,8 +3,9 @@
import com.baomidou.mybatisplus.core.metadata.IPage;
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.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;
@@ -17,8 +18,6 @@
import org.springblade.modules.property.service.IPropertyCompanyService;
import org.springblade.modules.property.service.IPropertyDistrictUserService;
import org.springblade.modules.property.vo.PropertyCompanyDistrictVO;
import org.springblade.modules.system.entity.User;
import org.springblade.modules.system.service.IUserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
@@ -50,26 +49,32 @@
   public IPage<PropertyCompanyDistrictVO> selectPropertyCompanyDistrictPage(IPage<PropertyCompanyDistrictVO> page, PropertyCompanyDistrictVO propertyCompanyDistrict) {
      // 物业 查询用户管理的小区
      String userRole = AuthUtil.getUserRole();
      if (userRole.contains("wygly") || userRole.contains("wyxmjl")) {
      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()));
         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);
         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<>());
         }
      }
      // 街道
      return page.setRecords(baseMapper.selectPropertyCompanyDistrictPage(page, propertyCompanyDistrict));
      List<PropertyCompanyDistrictVO> propertyCompanyDistrictVOS = baseMapper.selectPropertyCompanyDistrictPage(page, propertyCompanyDistrict);
      return page.setRecords(propertyCompanyDistrictVOS);
   }
   /**
@@ -80,18 +85,18 @@
    */
   @Override
   @Transactional(rollbackFor = Exception.class)
   public boolean saveOrUpdatePropertyCompanyDistrict(PropertyCompanyDistrictEntity propertyCompanyDistrict) throws Exception {
      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");
      }
   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);
         // bean.updateById(user);
         return updateById(propertyCompanyDistrict) && submitPropertyDistrictUser(propertyCompanyDistrict);
      } else {
         long count = count(Wrappers.<PropertyCompanyDistrictEntity>lambdaQuery()
@@ -99,9 +104,9 @@
            .eq(PropertyCompanyDistrictEntity::getPropertyCompanyId, propertyCompanyDistrict.getPropertyCompanyId())
            .eq(PropertyCompanyDistrictEntity::getIsDeleted, 0));
         if (count > 0) {
            throw new Exception("您已有该小区的合同,请勿重复添加!");
            throw new CustomException("您已有该小区的合同,请勿重复添加!");
         }
         bean.updateById(user);
         // bean.updateById(user);
         return save(propertyCompanyDistrict) && submitPropertyDistrictUser(propertyCompanyDistrict);
      }
   }
@@ -114,10 +119,11 @@
    */
   private boolean submitPropertyDistrictUser(PropertyCompanyDistrictEntity propertyCompanyDistrict) {
      List<Long> userIdList = Func.toLongList(propertyCompanyDistrict.getUserId());
      List<Long> arrayList = new ArrayList<Long>(userIdList);
      // 把项目经理也加到物业用户关联表
      userIdList.add(Long.valueOf(propertyCompanyDistrict.getPrincipal()));
      arrayList.add(Long.valueOf(propertyCompanyDistrict.getPrincipal()));
      List<PropertyDistrictUserEntity> propertyDistrictUserEntityList = new ArrayList<>();
      userIdList.forEach(userId -> {
      arrayList.forEach(userId -> {
         PropertyDistrictUserEntity propertyDistrictUserEntity = new PropertyDistrictUserEntity();
         propertyDistrictUserEntity.setUserId(userId);
         propertyDistrictUserEntity.setPropertyCompanyDistrictId(propertyCompanyDistrict.getId());