linwe
2024-06-18 9f1bbc41aa1ac91f1d7e7deeb85476796e95709c
src/main/java/org/springblade/modules/house/service/impl/HouseholdServiceImpl.java
@@ -23,6 +23,8 @@
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import liquibase.repackaged.org.apache.commons.lang3.StringUtils;
import org.apache.logging.log4j.util.Strings;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springblade.common.constant.CommonConstant;
import org.springblade.common.node.TreeIntegerNode;
import org.springblade.common.node.TreeStringNode;
@@ -62,6 +64,8 @@
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.stream.Collectors;
/**
@@ -72,6 +76,8 @@
 */
@Service
public class HouseholdServiceImpl extends ServiceImpl<HouseholdMapper, HouseholdEntity> implements IHouseholdService {
   private static Logger logger = LoggerFactory.getLogger(HouseholdServiceImpl.class);
   @Autowired
   private IUserHouseLabelService userHouseLabelService;
@@ -342,21 +348,94 @@
      int importNum = 0;
      int updateNum = 0;
      int errorNum = 0;
      // 创建一个固定大小的线程池
      int numberOfThreads = Runtime.getRuntime().availableProcessors(); // 使用可用处理器数量作为线程数
      ExecutorService executorService = Executors.newFixedThreadPool(numberOfThreads);
      // 创建Future列表来保存每个任务的结果
      List<Future<Map<String, String>>> futures = new ArrayList<>();
      for (ImportHouseholdExcel houseHoldExcel : data) {
         Callable<Map<String, String>> task = () -> importHouseHold(houseHoldExcel, houseService, isCovered, isTenant);
         futures.add(executorService.submit(task));
      }
      // 收集并打印结果
      for (Future<Map<String, String>> future : futures) {
         try {
            Map<String, String> stringStringMap = future.get();
            logger.info("结果:{}", stringStringMap);
            String result = stringStringMap.get("restult");
            switch (result) {
               case "1":
                  importNum++;
                  break;
               case "2":
                  updateNum++;
                  break;
               case "3":
                  errorNum++;
                  break;
               case "4":
                  importNum++;
                  updateNum++;
                  break;
               default:
                  errorNum++;
                  break;
            }
            // 获取并打印每个任务的结果
         } catch (InterruptedException | ExecutionException e) {
            errorNum++;
            logger.error("获取异常-----》", e);
         }
      }
      // 关闭线程池
      executorService.shutdown();
      try {
         if (!executorService.awaitTermination(60, TimeUnit.SECONDS)) {
            executorService.shutdownNow();
         }
      } catch (InterruptedException e) {
         executorService.shutdownNow();
      }
      StringBuilder builder = new StringBuilder("导入完成!");
      builder.append("其中本次表格共有 ").append(totalNum).append(" 条数据,")
         .append("成功导入 ").append(importNum).append(" 条数据。");
      if (updateNum > 0) {
         builder.append("其中有 ").append(updateNum).append(" 条数据已存在。");
      }
      if (errorNum > 0) {
         builder.append("共有 ").append(errorNum).append(" 条数据由于无姓名或门牌地址编码信息未导入!");
      }
      return builder.toString();
   }
   public Map<String, String> importHouseHold(ImportHouseholdExcel houseHoldExcel, IHouseService houseService, Boolean isCovered, String isTenant) {
      Map<String, String> objectObjectHashMap = new HashMap<>();
      try {
         HouseholdEntity householdEntity = Objects.requireNonNull(BeanUtil.copy(houseHoldExcel, HouseholdEntity.class));
         // 判断姓名,手机号
         if (Strings.isBlank(householdEntity.getName())
            || Strings.isBlank(householdEntity.getHouseCode())
         ) {
            errorNum++;
            continue;
         if (Strings.isBlank(householdEntity.getName()) || Strings.isBlank(householdEntity.getHouseCode())) {
            objectObjectHashMap.put("row", householdEntity.getName() + ":" + householdEntity.getPhoneNumber());
            objectObjectHashMap.put("restult", "3");
            return objectObjectHashMap;
         }
         // 去除手机号中间空格
         if (StringUtils.isNotBlank(householdEntity.getPhoneNumber())) {
            householdEntity.setPhoneNumber(householdEntity.getPhoneNumber().replaceAll(" ", ""));
         }
         // 查询库中是否已存在
         QueryWrapper<HouseholdEntity> wrapper = new QueryWrapper<>();
         wrapper.eq("house_code", householdEntity.getHouseCode())
            .eq("is_deleted", 0)
            .eq("name", householdEntity.getName());
         HouseholdEntity one = this.getOne(wrapper);
         // 判断是否租客
         if (StringUtils.isNotBlank(isTenant)) {
            wrapper.eq("relationship", 18);
         }
         List<HouseholdEntity> list = this.list(wrapper);
         // 与角色关系
         if (!Strings.isBlank(houseHoldExcel.getRelationship())) {
            householdEntity.setRelationship(Integer.parseInt(houseHoldExcel.getRelationship()));
@@ -369,26 +448,25 @@
         if (StringUtils.isNotBlank(isTenant)) {
            householdEntity.setRelationship(18);
            HouseRentalEntity houseRentalEntity = iHouseRentalService.getOne(Wrappers.<HouseRentalEntity>lambdaQuery()
               .eq(HouseRentalEntity::getHouseCode, householdEntity.getHouseCode()));
               .eq(HouseRentalEntity::getHouseCode, householdEntity.getHouseCode()).last("limit 1"));
            if (houseRentalEntity != null) {
               householdEntity.setHousingRentalId(houseRentalEntity.getId());
            } else {
               // 新建出租屋
               HouseRentalEntity houseRentalEntity1 = new HouseRentalEntity();
               houseRentalEntity1.setHouseCode(householdEntity.getHouseCode());
               houseRentalEntity1.setHouseStatus(1);
               houseRentalEntity1.setAuditStatus(0);
               houseRentalEntity1.setRentalUse(4);
               houseRentalEntity1.setTenantRelationship(1);
               houseRentalEntity1.setRentalTime(new Date());
               houseRentalEntity1.setDueTime(new Date());
               houseRentalEntity1.setTerminationTime(new Date());
               iHouseRentalService.save(houseRentalEntity1);
               householdEntity.setHousingRentalId(houseRentalEntity1.getId());
               // errorNum++;
               // continue;
               HouseRentalEntity rentalEntity = new HouseRentalEntity();
               rentalEntity.setHouseCode(householdEntity.getHouseCode());
               rentalEntity.setHouseStatus(1);
               rentalEntity.setAuditStatus(0);
               rentalEntity.setRentalUse(4);
               rentalEntity.setTenantRelationship(1);
               rentalEntity.setRentalTime(new Date());
               rentalEntity.setDueTime(new Date());
               rentalEntity.setTerminationTime(new Date());
               iHouseRentalService.save(rentalEntity);
               householdEntity.setHousingRentalId(rentalEntity.getId());
            }
         }
         /// 是否主要联系人
         if (!Strings.isBlank(houseHoldExcel.getIsPrimaryContact())) {
            householdEntity.setIsPrimaryContact(Integer.parseInt(houseHoldExcel.getIsPrimaryContact()));
@@ -442,44 +520,55 @@
            householdEntity.setMaritalStatus(Integer.parseInt(houseHoldExcel.getMaritalStatus()));
         }
         if (one==null){
         if (list.size() == 0) {
            // 新增
            this.save(householdEntity);
            importNum++;
            // importNum++;
            // 处理标签
            handleLabel(householdEntity,houseHoldExcel);
         }else {
            updateNum++;
            if (isCovered){
               // 覆盖更新
               householdEntity.setId(one.getId());
               // 更新
               updateById(householdEntity);
               importNum++;
               // 处理标签
               handleLabel(householdEntity,houseHoldExcel);
            handleLabel(householdEntity, houseHoldExcel);
            // return "1";
            objectObjectHashMap.put("row", householdEntity.getName() + ":" + householdEntity.getPhoneNumber());
            objectObjectHashMap.put("restult", "1");
            return objectObjectHashMap;
         } else {
            // updateNum++;
            if (isCovered) {
               for (HouseholdEntity one : list) {
                  // 覆盖更新
                  householdEntity.setId(one.getId());
                  // 更新
                  updateById(householdEntity);
                  // 处理标签
                  handleLabel(householdEntity, houseHoldExcel);
               }
               // importNum++;
               // return "4";
               objectObjectHashMap.put("row", householdEntity.getName() + ":" + householdEntity.getPhoneNumber());
               objectObjectHashMap.put("restult", "4");
               return objectObjectHashMap;
            }
            // return "2";
            objectObjectHashMap.put("row", householdEntity.getName() + ":" + householdEntity.getPhoneNumber());
            objectObjectHashMap.put("restult", "2");
            return objectObjectHashMap;
         }
      } catch (Exception e) {
         logger.error("异常导入------------》", e);
         // return "3";
         // objectObjectHashMap.put("row", householdEntity.getName() + ":" + householdEntity.getPhoneNumber());
         objectObjectHashMap.put("restult", "3");
         return objectObjectHashMap;
      }
      StringBuilder builder = new StringBuilder("导入完成!");
      builder.append("其中本次表格共有 ").append(totalNum).append(" 条数据,")
         .append("成功导入 ").append(importNum).append(" 条数据。");
      if (updateNum>0) {
         builder.append("其中有 ").append(updateNum).append(" 条数据已存在。");
      }
      if (errorNum>0){
         builder.append("共有 ").append(errorNum).append(" 条数据由于无姓名或门牌地址编码信息未导入!");
      }
      return builder.toString();
   }
   /**
    * 标签处理
    *
    * @param householdEntity
    * @param houseHoldExcel
    */
   public void handleLabel(HouseholdEntity householdEntity, ImportHouseholdExcel houseHoldExcel) {
      if (!Strings.isBlank(houseHoldExcel.getLabelName())){
      if (!Strings.isBlank(houseHoldExcel.getLabelName())) {
         // 查询对应的label集合信息
         List<UserHouseLabelEntity> userHouseLabelList = userHouseLabelService.getLabelList(houseHoldExcel.getLabelName());
         if (userHouseLabelList.size()>0){
@@ -811,4 +900,16 @@
      );
      return houseTenantExcels;
   }
   @Override
   public boolean saveOrUpdateHouseholdBatch(List<HouseholdVO> householdVOList) {
      if (householdVOList.size() == 0) {
         return false;
      }
      AtomicBoolean restult = new AtomicBoolean(false);
      householdVOList.forEach(item -> {
         restult.set(saveOrUpdateHousehold(item));
      });
      return restult.get();
   }
}