| | |
| | | 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; |
| | |
| | | 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; |
| | | |
| | | /** |
| | |
| | | */ |
| | | @Service |
| | | public class HouseholdServiceImpl extends ServiceImpl<HouseholdMapper, HouseholdEntity> implements IHouseholdService { |
| | | |
| | | private static Logger logger = LoggerFactory.getLogger(HouseholdServiceImpl.class); |
| | | |
| | | @Autowired |
| | | private IUserHouseLabelService userHouseLabelService; |
| | |
| | | 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())); |
| | |
| | | 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())); |
| | |
| | | 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){ |
| | |
| | | ); |
| | | 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(); |
| | | } |
| | | } |