package com.estate.expert.admin.initdata;
|
|
import com.estate.expert.admin.domain.standard.hierachy.Hierarchy;
|
import com.estate.expert.admin.domain.standard.measure.MeasureRule;
|
import com.estate.expert.admin.initdata.common.ImportUtil;
|
import com.estate.expert.admin.service.HierarchyService;
|
import com.estate.expert.admin.service.MeasureRuleService;
|
import com.estate.expert.admin.service.UserService;
|
import lombok.Builder;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.io.FileUtils;
|
import org.apache.poi.ss.usermodel.WorkbookFactory;
|
import org.junit.platform.commons.util.StringUtils;
|
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.UUID;
|
|
@Builder
|
@Slf4j
|
public class ImportMeasureRuleJob {
|
|
private UserService userService;
|
|
private HierarchyService hierarchyService;
|
|
private MeasureRuleService measureRuleService;
|
|
public void importMeasureRules() {
|
var hierarchy = hierarchyService.get(6);
|
|
loadDataFiles(hierarchy,"/实测实量-标准-规则管理");
|
}
|
|
private void loadDataFiles(Hierarchy hierarchy, String path){
|
var systemUser = userService.getSystemUser();
|
FileUtils.listFiles(new File(ImportUtil.docsRoot(path)), new String[]{"xlsx"}, true).forEach(f -> {
|
if (!f.getName().equals("data.xlsx"))
|
return;
|
|
var filePath = f.getAbsolutePath();
|
var hierarchyNamePath = filePath.substring(filePath.indexOf(path) + path.length(),
|
filePath.lastIndexOf("/"));
|
|
var hierarchyId = hierarchy.getId();
|
var hierarchyPath = hierarchy.getContent().getByNamePath(hierarchyNamePath).getPath();
|
|
var measureRules = loadMeasureRules(hierarchyId, hierarchyPath, filePath);
|
measureRuleService.addEntities(systemUser, measureRules);
|
});
|
}
|
|
private List<MeasureRule> loadMeasureRules(Integer hierarchyId, String hierarchyPath, String filePath){
|
var measureRules = new ArrayList<MeasureRule>();
|
|
try{
|
FileInputStream fis = new FileInputStream(filePath);
|
var workbook = WorkbookFactory.create(fis);
|
var sheet = workbook.getSheetAt(0);
|
int rowCount = sheet.getLastRowNum();
|
|
for (int i = 0; i <= rowCount; i++) {
|
var row = sheet.getRow(i);
|
var name = row.getCell(0).getStringCellValue();
|
if(StringUtils.isBlank(name)){
|
continue;
|
}
|
var measureRule = new MeasureRule();
|
measureRule.setHierarchyId(hierarchyId);
|
measureRule.setHierarchyPath(hierarchyPath);
|
measureRule.setName(UUID.randomUUID().toString());
|
measureRule.setDescription(row.getCell(0).getStringCellValue());
|
measureRule.setMeasurePoint(Integer.parseInt(row.getCell(1).getStringCellValue()));
|
measureRule.setMeasureType(row.getCell(2).getStringCellValue());
|
measureRule.setMeasureExpr1(row.getCell(3).getStringCellValue());
|
measureRule.setMeasureExpr2(row.getCell(4).getStringCellValue());
|
measureRules.add(measureRule);
|
}
|
}catch (Exception e){
|
log.error("导入实测实量规则数据失败", e);
|
}
|
|
return measureRules;
|
}
|
}
|
