forked from luyan/epmet-cloud-lingshan
zxc
3 years ago
7 changed files with 368 additions and 5 deletions
@ -0,0 +1,113 @@ |
|||||
|
package com.epmet.model; |
||||
|
|
||||
|
import com.alibaba.excel.context.AnalysisContext; |
||||
|
import com.alibaba.excel.event.AnalysisEventListener; |
||||
|
import com.epmet.commons.tools.constant.NumConstant; |
||||
|
import com.epmet.commons.tools.enums.DictTypeEnum; |
||||
|
import com.epmet.commons.tools.exception.EpmetException; |
||||
|
import com.epmet.commons.tools.redis.common.bean.AgencyInfoCache; |
||||
|
import com.epmet.commons.tools.utils.ConvertUtils; |
||||
|
import com.epmet.commons.tools.utils.Result; |
||||
|
import com.epmet.commons.tools.validator.ValidatorUtils; |
||||
|
import com.epmet.dto.result.IcCoverageCategoryDictListResultDTO; |
||||
|
import com.epmet.entity.IcDangerousChemicalsEntity; |
||||
|
import com.epmet.feign.EpmetAdminOpenFeignClient; |
||||
|
import com.epmet.service.CoverageService; |
||||
|
import com.epmet.service.IcDangerousChemicalsService; |
||||
|
import lombok.Data; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.apache.commons.collections4.CollectionUtils; |
||||
|
import org.apache.commons.collections4.MapUtils; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.HashMap; |
||||
|
import java.util.List; |
||||
|
import java.util.Map; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
/** |
||||
|
* @Author zxc |
||||
|
* @DateTime 2022/6/20 10:31 |
||||
|
* @DESC |
||||
|
*/ |
||||
|
@Data |
||||
|
@Slf4j |
||||
|
public class DangerousChemicalsImportListener extends AnalysisEventListener<DangerousChemicalsModel> { |
||||
|
|
||||
|
private IcDangerousChemicalsService icDangerousChemicalsService; |
||||
|
private EpmetAdminOpenFeignClient adminOpenFeignClient; |
||||
|
private CoverageService coverageService; |
||||
|
private AgencyInfoCache agencyInfo; |
||||
|
|
||||
|
private List<DangerousChemicalsModel.RowRemarkMessage> errorRows = new ArrayList<>(); |
||||
|
private List<DangerousChemicalsModel.RowRemarkMessage> otherRows = new ArrayList<>(); |
||||
|
private List<IcDangerousChemicalsEntity> insertList = new ArrayList<>(); |
||||
|
|
||||
|
public DangerousChemicalsImportListener(AgencyInfoCache agencyInfo, IcDangerousChemicalsService icDangerousChemicalsService, CoverageService coverageService,EpmetAdminOpenFeignClient adminOpenFeignClient){ |
||||
|
this.agencyInfo = agencyInfo; |
||||
|
this.icDangerousChemicalsService = icDangerousChemicalsService; |
||||
|
this.coverageService = coverageService; |
||||
|
this.adminOpenFeignClient = adminOpenFeignClient; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void invoke(DangerousChemicalsModel data, AnalysisContext context) { |
||||
|
IcDangerousChemicalsEntity e = ConvertUtils.sourceToTarget(data, IcDangerousChemicalsEntity.class); |
||||
|
e.setSourceType("import"); |
||||
|
e.setAgencyId(agencyInfo.getId()); |
||||
|
e.setCustomerId(agencyInfo.getCustomerId()); |
||||
|
e.setAgencyIdPath(agencyInfo.getPids().equals(NumConstant.ZERO_STR) || agencyInfo.getPids().equals("") ? agencyInfo.getId() : agencyInfo.getPids().concat(":").concat(agencyInfo.getId())); |
||||
|
insertList.add(e); |
||||
|
if (insertList.size() == NumConstant.ONE_HUNDRED){ |
||||
|
execPersist(); |
||||
|
} |
||||
|
|
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void doAfterAllAnalysed(AnalysisContext context) { |
||||
|
execPersist(); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Desc: 数据库插入 |
||||
|
*/ |
||||
|
private void execPersist() { |
||||
|
try { |
||||
|
if (CollectionUtils.isNotEmpty(insertList)) { |
||||
|
//危化品种类字典
|
||||
|
Result<Map<String, String>> statusRes = adminOpenFeignClient.dictMap(DictTypeEnum.IC_DANGER_TYPE.getCode()); |
||||
|
if (!statusRes.success()){ |
||||
|
throw new EpmetException("获取IC_DANGER_TYPE字典表失败"); |
||||
|
} |
||||
|
Map<String, String> statusMap = statusRes.success() && MapUtils.isNotEmpty(statusRes.getData()) ? statusRes.getData() : new HashMap<>(16); |
||||
|
//企业类别字典数据
|
||||
|
List<IcCoverageCategoryDictListResultDTO> dictList = coverageService.dictMap(agencyInfo.getCustomerId(), "dangerous_chemicals"); |
||||
|
Map<String, String> dictMap = dictList.stream().collect(Collectors.toMap(IcCoverageCategoryDictListResultDTO::getCategoryKey, IcCoverageCategoryDictListResultDTO::getCategoryName)); |
||||
|
insertList.forEach(i -> { |
||||
|
statusMap.forEach((k,v) -> { |
||||
|
if (i.getDangerTypeName().equals(v)){ |
||||
|
i.setDangerType(k); |
||||
|
} |
||||
|
}); |
||||
|
dictMap.forEach((k,v) -> { |
||||
|
if (i.getCategoryName().equals(v)){ |
||||
|
i.setCategory(k); |
||||
|
} |
||||
|
}); |
||||
|
}); |
||||
|
icDangerousChemicalsService.insertBatch(insertList); |
||||
|
} |
||||
|
} finally { |
||||
|
insertList.clear(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 获取错误行 |
||||
|
* @return |
||||
|
*/ |
||||
|
public List<DangerousChemicalsModel.RowRemarkMessage> getErrorRows() { |
||||
|
return errorRows; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,56 @@ |
|||||
|
package com.epmet.model; |
||||
|
|
||||
|
import com.alibaba.excel.annotation.ExcelProperty; |
||||
|
import com.alibaba.excel.annotation.write.style.ColumnWidth; |
||||
|
import com.epmet.commons.tools.constant.NumConstant; |
||||
|
import lombok.Data; |
||||
|
import org.hibernate.validator.constraints.Length; |
||||
|
|
||||
|
import javax.validation.constraints.NotBlank; |
||||
|
|
||||
|
/** |
||||
|
* @Author zxc |
||||
|
* @DateTime 2022/6/20 10:33 |
||||
|
* @DESC |
||||
|
*/ |
||||
|
@Data |
||||
|
public class DangerousChemicalsModel { |
||||
|
|
||||
|
@NotBlank(message = "企业名称为必填项") |
||||
|
@Length(max = 50,message = "企业名称长度超出50字限制") |
||||
|
@ExcelProperty(value = "企业名称") |
||||
|
private String name; |
||||
|
|
||||
|
@NotBlank(message = "企业类型为必填项") |
||||
|
@ExcelProperty(value = "企业类型") |
||||
|
private String categoryName; |
||||
|
|
||||
|
@ExcelProperty(value = "周边安全间距（公里）") |
||||
|
private String safeDistance = NumConstant.ZERO_STR; |
||||
|
|
||||
|
@ExcelProperty(value = "危化品种类") |
||||
|
private String dangerTypeName; |
||||
|
|
||||
|
@ExcelProperty(value = "负责人") |
||||
|
private String principalName; |
||||
|
|
||||
|
@Length(max = 50,message = "联系电话长度超出50字限制") |
||||
|
@ExcelProperty(value = "联系电话") |
||||
|
private String principalMobile; |
||||
|
|
||||
|
@Length(max = 50,message = "经营地址长度超出50字限制") |
||||
|
@ExcelProperty(value = "经营地址") |
||||
|
private String address; |
||||
|
|
||||
|
@ExcelProperty(value = "备注") |
||||
|
private String remark; |
||||
|
|
||||
|
@Data |
||||
|
public static class RowRemarkMessage { |
||||
|
|
||||
|
@ColumnWidth(60) |
||||
|
@ExcelProperty("错误信息") |
||||
|
private String errorInfo; |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue