Java poi导入合并单元格的Excel数据,复杂树形结构,处理重复数据

导入一个井口-段塞-堵剂,三层树形结构的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);
    }
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值