forked from rongchao/epmet-cloud-rizhao
yinzuomei
2 years ago
8 changed files with 390 additions and 8 deletions
@ -0,0 +1,120 @@ |
|||||
|
package com.epmet.excel.handler; |
||||
|
|
||||
|
import com.alibaba.excel.context.AnalysisContext; |
||||
|
import com.alibaba.excel.read.listener.ReadListener; |
||||
|
import com.epmet.commons.tools.exception.EpmetException; |
||||
|
import com.epmet.commons.tools.exception.ExceptionUtils; |
||||
|
import com.epmet.commons.tools.exception.ValidateException; |
||||
|
import com.epmet.commons.tools.utils.ConvertUtils; |
||||
|
import com.epmet.commons.tools.utils.ObjectUtil; |
||||
|
import com.epmet.commons.tools.validator.ValidatorUtils; |
||||
|
import com.epmet.entity.IcPropertyManagementEntity; |
||||
|
import com.epmet.excel.yt.IcPropertyManagementImportExcelData; |
||||
|
import com.epmet.service.impl.IcPropertyManagementServiceImpl; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.apache.commons.collections4.CollectionUtils; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
|
||||
|
/** |
||||
|
* @Description |
||||
|
* @Author yzm |
||||
|
* @Date 2023/5/8 14:23 |
||||
|
*/ |
||||
|
@Slf4j |
||||
|
public class IcPropertyManagementImportListener implements ReadListener<IcPropertyManagementImportExcelData> { |
||||
|
|
||||
|
// 最大条数阈值
|
||||
|
public static final int MAX_THRESHOLD = 200; |
||||
|
private String currentCustomerId; |
||||
|
private IcPropertyManagementServiceImpl propertyManagementService; |
||||
|
// 错误项列表
|
||||
|
private List<IcPropertyManagementImportExcelData.ErrorRow> errorRows = new ArrayList<>(); |
||||
|
// 要插入的数据
|
||||
|
private List<IcPropertyManagementEntity> insertDatas = new ArrayList<>(); |
||||
|
private List<IcPropertyManagementEntity> updateDatas = new ArrayList<>(); |
||||
|
|
||||
|
public IcPropertyManagementImportListener(String customerId, IcPropertyManagementServiceImpl propertyManagementService) { |
||||
|
this.currentCustomerId = customerId; |
||||
|
this.propertyManagementService = propertyManagementService; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void invoke(IcPropertyManagementImportExcelData data, AnalysisContext analysisContext) { |
||||
|
try { |
||||
|
// log.warn("有数据吗？"+JSON.toJSONString(data));
|
||||
|
// 不能为空先校验数据
|
||||
|
ValidatorUtils.validateEntity(data); |
||||
|
// 去除空格
|
||||
|
ObjectUtil.objectToTrim(data); |
||||
|
//物业名称唯一
|
||||
|
IcPropertyManagementEntity origin=propertyManagementService.getByName(currentCustomerId, data.getName()); |
||||
|
|
||||
|
IcPropertyManagementEntity propertyManagementEntity = ConvertUtils.sourceToTarget(data, IcPropertyManagementEntity.class); |
||||
|
propertyManagementEntity.setCustomerId(currentCustomerId); |
||||
|
if (null != origin) { |
||||
|
origin.setContactMobile(data.getContactMobile()); |
||||
|
origin.setContactName(data.getContactName()); |
||||
|
insertDatas.add(origin); |
||||
|
} else { |
||||
|
insertDatas.add(propertyManagementEntity); |
||||
|
} |
||||
|
|
||||
|
if (insertDatas.size() == MAX_THRESHOLD) { |
||||
|
execPersist(); |
||||
|
} |
||||
|
if (updateDatas.size() == MAX_THRESHOLD) { |
||||
|
execPersist(); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
String errorMsg = null; |
||||
|
if (e instanceof ValidateException) { |
||||
|
errorMsg = ((ValidateException) e).getMsg(); |
||||
|
} else if (e instanceof EpmetException) { |
||||
|
errorMsg = ((EpmetException) e).getInternalMsg(); |
||||
|
} else { |
||||
|
errorMsg = "未知错误"; |
||||
|
log.error("【物业管理表ic_property_management导入】出错:{}", ExceptionUtils.getErrorStackTrace(e)); |
||||
|
} |
||||
|
IcPropertyManagementImportExcelData.ErrorRow errorRow = ConvertUtils.sourceToTarget(data, IcPropertyManagementImportExcelData.ErrorRow.class); |
||||
|
errorRow.setErrorInfo(errorMsg); |
||||
|
errorRows.add(errorRow); |
||||
|
} |
||||
|
|
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void doAfterAllAnalysed(AnalysisContext analysisContext) { |
||||
|
// 最后几条达不到阈值，这里必须再调用一次
|
||||
|
execPersist(); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 执行持久化 |
||||
|
*/ |
||||
|
private void execPersist() { |
||||
|
try { |
||||
|
if (CollectionUtils.isNotEmpty(insertDatas)) { |
||||
|
propertyManagementService.insertBatch(insertDatas); |
||||
|
} |
||||
|
|
||||
|
if (CollectionUtils.isNotEmpty(updateDatas)) { |
||||
|
propertyManagementService.updateBatchById(updateDatas); |
||||
|
} |
||||
|
} finally { |
||||
|
insertDatas.clear(); |
||||
|
updateDatas.clear(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 获取错误行 |
||||
|
* |
||||
|
* @return |
||||
|
*/ |
||||
|
public List<IcPropertyManagementImportExcelData.ErrorRow> getErrorRows() { |
||||
|
return errorRows; |
||||
|
} |
||||
|
} |
||||
|
|
||||
@ -0,0 +1,58 @@ |
|||||
|
package com.epmet.excel.yt; |
||||
|
|
||||
|
import com.alibaba.excel.annotation.ExcelProperty; |
||||
|
import com.alibaba.excel.annotation.write.style.ColumnWidth; |
||||
|
import lombok.Data; |
||||
|
|
||||
|
/** |
||||
|
* @Description |
||||
|
* @Author yzm |
||||
|
* @Date 2023/5/8 14:24 |
||||
|
*/ |
||||
|
@Data |
||||
|
public class IcPropertyManagementImportExcelData { |
||||
|
/** |
||||
|
* 物业名称 |
||||
|
*/ |
||||
|
@ExcelProperty(value = "物业名称") |
||||
|
private String name; |
||||
|
|
||||
|
/** |
||||
|
* 烟台需求：物业联系人姓名 |
||||
|
*/ |
||||
|
@ExcelProperty(value = "物业联系人") |
||||
|
private String contactName; |
||||
|
|
||||
|
/** |
||||
|
* 烟台需求：物业联系人电话 |
||||
|
*/ |
||||
|
@ExcelProperty(value = "联系电话") |
||||
|
private String contactMobile; |
||||
|
|
||||
|
@Data |
||||
|
public static class ErrorRow { |
||||
|
/** |
||||
|
* 物业名称 |
||||
|
*/ |
||||
|
@ExcelProperty(value = "物业名称") |
||||
|
private String name; |
||||
|
|
||||
|
/** |
||||
|
* 烟台需求：物业联系人姓名 |
||||
|
*/ |
||||
|
@ExcelProperty(value = "物业联系人") |
||||
|
private String contactName; |
||||
|
|
||||
|
/** |
||||
|
* 烟台需求：物业联系人电话 |
||||
|
*/ |
||||
|
@ExcelProperty(value = "联系电话") |
||||
|
private String contactMobile; |
||||
|
|
||||
|
|
||||
|
@ColumnWidth(60) |
||||
|
@ExcelProperty("错误信息") |
||||
|
private String errorInfo; |
||||
|
} |
||||
|
} |
||||
|
|
||||
Loading…
Reference in new issue