linwe
2023-11-11 4cc3da53fca83e9212511c980acb6799ec346565
src/main/java/org/springblade/modules/doorplateAddress/service/impl/DoorplateAddressServiceImpl.java
@@ -16,25 +16,20 @@
 */
package org.springblade.modules.doorplateAddress.service.impl;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import liquibase.repackaged.org.apache.commons.lang3.StringUtils;
import org.apache.poi.hssf.record.DVALRecord;
import org.jetbrains.annotations.NotNull;
import org.springblade.common.constant.DictConstant;
import org.springblade.common.node.TreeNode;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.utils.StringUtil;
import org.springblade.modules.doorplateAddress.entity.DoorplateAddressEntity;
import org.springblade.modules.doorplateAddress.mapper.DoorplateAddressMapper;
import org.springblade.modules.doorplateAddress.service.IDoorplateAddressService;
import org.springblade.modules.doorplateAddress.vo.DoorplateAddressVOTree;
import org.springblade.modules.doorplateAddress.vo.DoorplateAddressVO;
import org.springblade.modules.doorplateAddress.vo.DoorplateAddressVOTree;
import org.springblade.modules.doorplateAddress.vo.FuncNode;
import org.springblade.modules.house.service.IHouseRentalService;
import org.springblade.modules.house.service.IHouseholdLabelService;
import org.springblade.modules.house.service.IHouseholdService;
import org.springblade.modules.house.vo.HouseParam;
import org.springblade.modules.house.vo.HouseRentalVO;
@@ -42,7 +37,6 @@
import org.springblade.modules.place.service.IPlaceService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import java.util.*;
import java.util.stream.Collectors;
@@ -116,9 +110,28 @@
   private Object getInhabitantInfo(List<TreeNode> list) {
      // 查询房屋集合信息
      List<TreeNode> houseNodeList = householdService.selectHouseNodeList(AuthUtil.getUserId());
      for (TreeNode treeNode : houseNodeList) {
         // 判断房屋类型类型
         if (treeNode.getDoorplateType().equals(DictConstant.SMALL_DOORPLATE) ||
            (treeNode.getDoorplateType().equals(DictConstant.centre_DOORPLATE) &&
               treeNode.getAddressLevel().equals(1))) {
            treeNode.setAddressType(2);
         } else {
            treeNode.setAddressType(1);
         }
      }
      // 查询场所集合信息
      List<TreeNode> placeNodeList = placeService.selectPlaceNodeList(AuthUtil.getUserId());
      if (houseNodeList.size()>0 && placeNodeList.size()>0){
      for (TreeNode treeNode : placeNodeList) {
         if (treeNode.getDoorplateType().equals(DictConstant.SMALL_DOORPLATE) ||
            (treeNode.getDoorplateType().equals(DictConstant.centre_DOORPLATE) &&
               treeNode.getAddressLevel().equals(1))) {
            treeNode.setAddressType(2);
         } else {
            treeNode.setAddressType(3);
         }
      }
      if (houseNodeList.size() > 0 && placeNodeList.size() > 0) {
         // 合并
         TreeNode houseNode = new TreeNode();
         houseNode.setName("房屋");
@@ -408,4 +421,16 @@
      // 返回
      return doorplateAddress;
   }
   @Override
   public Object getHouseType(String code) {
      DoorplateAddressEntity doorplateAddressEntity = baseMapper.selectOne(Wrappers.<DoorplateAddressEntity>lambdaQuery().eq(DoorplateAddressEntity::getAddressCode, code));
      // 判断房屋信息是否住宅
      if (doorplateAddressEntity.getDoorplateType().equals(DictConstant.SMALL_DOORPLATE) ||
         (doorplateAddressEntity.getDoorplateType().equals(DictConstant.centre_DOORPLATE) &&
            doorplateAddressEntity.getAddressLevel().equals(1))) {
         return 1;
      }
      return 0;
   }
}