forked from rongchao/epmet-cloud-rizhao
yinzuomei
2 years ago
8 changed files with 408 additions and 8 deletions
@ -0,0 +1,110 @@ |
|||
package com.epmet.excel.handler; |
|||
|
|||
import com.alibaba.excel.context.AnalysisContext; |
|||
import com.alibaba.excel.read.listener.ReadListener; |
|||
import com.epmet.commons.tools.constant.NumConstant; |
|||
import com.epmet.commons.tools.constant.StrConstant; |
|||
import com.epmet.commons.tools.exception.EpmetException; |
|||
import com.epmet.commons.tools.exception.ExceptionUtils; |
|||
import com.epmet.commons.tools.utils.ConvertUtils; |
|||
import com.epmet.commons.tools.validator.ValidatorUtils; |
|||
import com.epmet.entity.CustomerAgencyEntity; |
|||
import com.epmet.entity.CustomerDepartmentEntity; |
|||
import com.epmet.excel.yt.DeptImportExcelDTO; |
|||
import com.epmet.service.CustomerDepartmentService; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.apache.commons.collections4.CollectionUtils; |
|||
import org.apache.commons.lang3.StringUtils; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.List; |
|||
|
|||
/** |
|||
* @Description |
|||
* @Author yzm |
|||
* @Date 2023/4/26 16:11 |
|||
*/ |
|||
@Slf4j |
|||
public class DeptExcelImportListener implements ReadListener<DeptImportExcelDTO> { |
|||
//最大条数阈值
|
|||
public static final int MAX_THRESHOLD = 200; |
|||
private String customerId; |
|||
private String staffId; |
|||
private CustomerAgencyEntity agencyEntity; |
|||
/** |
|||
* 当前操作用户 |
|||
*/ |
|||
//要插入的数据
|
|||
private List<CustomerDepartmentEntity> insertDatas = new ArrayList<>(); |
|||
//错误项列表
|
|||
private List<DeptImportExcelDTO.ErrorRow> errorRows = new ArrayList<>(); |
|||
private CustomerDepartmentService departmentService; |
|||
public DeptExcelImportListener(String loginUserCustomerId, String loginUserId, CustomerAgencyEntity agencyEntity, CustomerDepartmentService departmentService) { |
|||
this.customerId = loginUserCustomerId; |
|||
this.staffId = loginUserId; |
|||
this.agencyEntity = agencyEntity; |
|||
this.departmentService = departmentService; |
|||
} |
|||
|
|||
@Override |
|||
public void invoke(DeptImportExcelDTO data, AnalysisContext analysisContext) { |
|||
try { |
|||
ValidatorUtils.validateEntity(data); |
|||
//当前组织下，是否存在该部门
|
|||
departmentService.checkUnqiueName(agencyEntity.getId(),data.getDepartmentName(),null); |
|||
CustomerDepartmentEntity entity = ConvertUtils.sourceToTarget(data, CustomerDepartmentEntity.class); |
|||
entity.setCustomerId(customerId); |
|||
entity.setAgencyId(agencyEntity.getId()); |
|||
entity.setCreatedBy(staffId); |
|||
entity.setUpdatedBy(staffId); |
|||
entity.setTotalUser(NumConstant.ZERO); |
|||
entity.setAreaCode(StringUtils.isNotBlank(agencyEntity.getAreaCode())?agencyEntity.getAreaCode(): StrConstant.EPMETY_STR); |
|||
insertDatas.add(entity); |
|||
if (insertDatas.size() == MAX_THRESHOLD) { |
|||
execPersist(); |
|||
} |
|||
} catch (Exception e) { |
|||
String errorMsg = null; |
|||
if(e instanceof EpmetException){ |
|||
errorMsg=((EpmetException) e).getMsg(); |
|||
}else { |
|||
errorMsg = "未知错误"; |
|||
log.error("【部门导入】出错:{}", ExceptionUtils.getErrorStackTrace(e)); |
|||
} |
|||
DeptImportExcelDTO.ErrorRow errorRow = ConvertUtils.sourceToTarget(data,DeptImportExcelDTO.ErrorRow.class); |
|||
errorRow.setErrorInfo(errorMsg); |
|||
errorRows.add(errorRow); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void doAfterAllAnalysed(AnalysisContext context) { |
|||
// 最后几条达不到阈值，这里必须再调用一次
|
|||
execPersist(); |
|||
} |
|||
|
|||
/** |
|||
* 执行持久化 |
|||
*/ |
|||
private void execPersist() { |
|||
try { |
|||
if (CollectionUtils.isNotEmpty(insertDatas)) { |
|||
departmentService.insertBatch(insertDatas); |
|||
} |
|||
|
|||
} finally { |
|||
insertDatas.clear(); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 获取错误行 |
|||
* |
|||
* @return |
|||
*/ |
|||
public List<DeptImportExcelDTO.ErrorRow> getErrorRows() { |
|||
return errorRows; |
|||
} |
|||
|
|||
} |
|||
|
|||
@ -0,0 +1,72 @@ |
|||
package com.epmet.excel.yt; |
|||
|
|||
import com.alibaba.excel.annotation.ExcelProperty; |
|||
import com.alibaba.excel.annotation.write.style.ColumnWidth; |
|||
import lombok.Data; |
|||
import org.hibernate.validator.constraints.Length; |
|||
|
|||
import javax.validation.constraints.NotBlank; |
|||
|
|||
/** |
|||
* @Description |
|||
* @Author yzm |
|||
* @Date 2023/4/26 15:59 |
|||
*/ |
|||
@Data |
|||
public class DeptImportExcelDTO { |
|||
/** |
|||
* 部门名称 |
|||
*/ |
|||
@NotBlank(message = "部门名称不能为空") |
|||
@Length(max = 50, message = "部门名称不能超过50个字") |
|||
@ExcelProperty(value = "*部门名称") |
|||
private String departmentName; |
|||
|
|||
/** |
|||
* 组织编码 |
|||
*/ |
|||
@ExcelProperty(value = "组织编码") |
|||
private String code; |
|||
/** |
|||
* 负责人 |
|||
*/ |
|||
@ExcelProperty(value = "联系人") |
|||
private String contacts; |
|||
/** |
|||
* 联系电话 |
|||
*/ |
|||
@ExcelProperty(value = "联系电话") |
|||
private String mobile; |
|||
|
|||
@Data |
|||
public static class ErrorRow { |
|||
@ColumnWidth(50) |
|||
@ExcelProperty(value = "*部门名称") |
|||
private String departmentName; |
|||
|
|||
/** |
|||
* 组织编码 |
|||
*/ |
|||
@ColumnWidth(20) |
|||
@ExcelProperty(value = "组织编码") |
|||
private String code; |
|||
/** |
|||
* 负责人 |
|||
*/ |
|||
@ColumnWidth(20) |
|||
@ExcelProperty(value = "联系人") |
|||
private String contacts; |
|||
|
|||
/** |
|||
* 联系电话 |
|||
*/ |
|||
@ColumnWidth(20) |
|||
@ExcelProperty(value = "联系电话") |
|||
private String mobile; |
|||
|
|||
@ColumnWidth(50) |
|||
@ExcelProperty("错误信息") |
|||
private String errorInfo; |
|||
} |
|||
} |
|||
|
|||
Binary file not shown.
Loading…
Reference in new issue