导入一个井口-段塞-堵剂,三层树形结构的Excel数据,每一个井口Well可能包含一个或多个段塞Slug,每一个段塞可分为一个或多个作业阶段,对应每个作业阶段需要添加一个或多个堵剂Material,如图所示,Excel数据中可能存在合并单元格的情况

理想情况下我们希望的Excel数据输入如下图所示,代码也将允许重复数据输入并进行存储。代码最终的存储结构将会是一个List<ExcelImportBean>的井口列表形式,每一个ExcelImportBean中将包含段塞列表List<Slug>,每一个段塞Slug中将包含堵剂列表List<Material>

核心的Excel中数据解析代码片段如下,这里输入的importList是原始的Excel数据列表,循环读取每一条数据,代码的核心是如果遇到井口为空或重复,证明仍是同一条井口的数据,则只更新段塞列表,沿用之前段塞保存过的井口信息;对段塞同理,如果作业阶段为空或重复,证明仍是同一个段塞的数据,则只更新堵剂列表,沿用之前保存过的段塞信息。如此往复:
ExcelImportBean wellData = new ExcelImportBean();
Slug slugData = new Slug();
Material materialData = new Material();
List<Slug> slugList = new ArrayList<>();
List<Material> materialList = new ArrayList<>();
String wellName = "";
Integer slugNode = 0;
Integer taskType = 0;
// importList存入的数据中含有很多空白项或重复项
// 需要根据井口或段塞合并,遍历importList,重新整理成井口、段塞结构存储
// 原本importList的结构与excel导入完全一致,一个数据对应一行
// 现在按井口存储数据,段塞存为slugList的格式
for (ExcelImportBean excelRow : importList) {
// 判断井口是否为空或重复
// 井口为空或重复的情况下,只更新slugList,沿用之前保存的井口信息,不需要再更新wellData
if (excelRow.getCode() == null || excelRow.getCode().equals(wellName)) {
// 判断段塞作业阶段是否为空或重复
// 段塞为空或重复的情况下,只更新materialList,沿用之前保存的段塞信息
if (excelRow.getNode() == null || excelRow.getNode().equals(slugNode)) {
if (excelRow.getTaskType() == null || excelRow.getTaskType().equals(taskType)) {
// 只存储堵剂信息,只更新materialList
materialData = new Material();
materialData.setCommonName(excelRow.getCommonName());
materialData.setDesignConcentration(excelRow.getDesignConcentration());
materialData.setDesignMaterial(excelRow.getDesignMaterial());
materialList.add(materialData);
slugData.setDefaultMaterials(materialList);
}
} else {
// 每个段塞的第一条数据,有段塞全部信息
// 重置段塞和堵剂
slugData = new Slug();
materialData = new Material();
materialList = new ArrayList<>();
if (excelRow.getTaskType() == null && taskType > 0) {
excelRow.setTaskType(taskType);
}
slugData.setTaskType(excelRow.getTaskType());
slugData.setNode(excelRow.getNode());
slugData.setVolume(excelRow.getVolume());
materialData.setCommonName(excelRow.getCommonName());
materialData.setDesignConcentration(excelRow.getDesignConcentration());
materialData.setDesignMaterial(excelRow.getDesignMaterial());
// 更新堵剂列表
materialList.add(materialData);
slugData.setDefaultMaterials(materialList);
// 更新段塞列表
slugList.add(slugData);
}
} else {
// 段塞列表是否有数据,如果有则把数据保存起来
if (slugList.size() != 0) {
wellData.setSlugList(slugList);
newList.add(wellData);
}
// 每个井口的第一条数据,有全部信息
// 重置井口和段塞列表
wellData = new WellProImportBean();
slugData = new Slug();
slugList = new ArrayList<>();
wellData.setCode(excelRow.getCode());
wellData.setBlock(excelRow.getBlock());
wellName = wellData.getCode();
// 重置堵剂
materialData = new Material();
materialList = new ArrayList<>();
slugData.setTaskType(excelRow.getTaskType());
slugData.setNode(excelRow.getNode());
slugData.setVolume(excelRow.getVolume());
materialData.setCommonName(excelRow.getCommonName());
materialData.setDesignConcentration(excelRow.getDesignConcentration());
materialData.setDesignMaterial(excelRow.getDesignMaterial());
// 更新堵剂列表
materialList.add(materialData);
slugData.setDefaultMaterials(materialList);
// 更新段塞列表
slugList.add(slugData);
wellData.setSlugList(slugList);
}
}
766

被折叠的 条评论
为什么被折叠?



