快捷導(dǎo)航

Java詳解如何將excel數(shù)據(jù)轉(zhuǎn)為樹(shù)形

更新時(shí)間：2022年08月12日 09:30:48 作者：Q?C

在平常的辦公工作中，excel數(shù)據(jù)的操作是最常見(jiàn)的需求，今天就來(lái)看一下通過(guò)Java如何來(lái)實(shí)現(xiàn)將excel數(shù)據(jù)轉(zhuǎn)為樹(shù)形，感興趣的朋友可以了解下

前言

今天收到一個(gè)導(dǎo)入的任務(wù),要求將excel數(shù)據(jù)保存到數(shù)據(jù)庫(kù)中,不同于普通的導(dǎo)入,這個(gè)導(dǎo)入的數(shù)據(jù)是一個(gè)樹(shù)形結(jié)構(gòu),如下圖:

通過(guò)觀察數(shù)據(jù)中的層級(jí)列我們發(fā)現(xiàn)表格數(shù)據(jù)由2棵樹(shù)組成,分別是第3,4,5,6,7,8,9,10,11和12,13,14,15,16,17,18,它們由0作樹(shù)的根節(jié)點(diǎn),1為0的子節(jié)點(diǎn),2為相鄰1的子節(jié)點(diǎn),由此得出第一顆樹(shù)的結(jié)構(gòu)為:

拆分原始數(shù)據(jù)

1.創(chuàng)建實(shí)體類

創(chuàng)建vo接收解析數(shù)據(jù),在這里,我們只關(guān)心層級(jí)屬性

	@Excel(name = "層級(jí)")
    private String hierarchy;
    @Excel(name = "物料編碼")
    private String materialCode;
    @Excel(name = "物料名稱")
    private String materialName;
    @Excel(name = "基礎(chǔ)數(shù)量")
    private BigDecimal materialNum;
    @Excel(name = "使用數(shù)量")
    private BigDecimal useAmount;
    @Excel(name = "BOM版本")
    private String version;
    @Excel(name = "默認(rèn)BOM")
    private String isDefaults;

2.處理數(shù)據(jù)

將數(shù)據(jù)源拆分為若干棵樹(shù)的數(shù)據(jù)集

代碼如下（示例）：

/**
     * 將集合對(duì)象按指定元素分割存儲(chǔ)
     *
     * @param materialVos 原始集合
     * @param s 分割元素(這里是當(dāng)集合對(duì)象層級(jí)為0時(shí)則分割,也就是樹(shù)的根節(jié)點(diǎn)為0)
     * @return 每棵樹(shù)的結(jié)果集
     */
    private List<List<MatMaterialBomImportVo>> subsection(List<MatMaterialBomImportVo> materialVos, String s) {
        List<List<MatMaterialBomImportVo>> segmentedData = new ArrayList<>();
        if (materialVos != null) {
            //獲取指定元素的數(shù)量,判斷出最終將拆分為多少段
            List<MatMaterialBomImportVo> collect = materialVos.stream().filter(bom -> s.equals(bom.getHierarchy())).collect(Collectors.toList());
            int count = 0;
            for (int i = 0; i < collect.size(); i++) {
                List<MatMaterialBomImportVo> bomImportVo = new ArrayList<>();
                boolean num = false;
                //遍歷數(shù)據(jù)源
                for (; count < materialVos.size(); count++) {
                    //第一個(gè)必然為樹(shù)的根節(jié)點(diǎn),直接獲取并跳過(guò)
                    if (count == 0) {
                        bomImportVo.add(materialVos.get(count));
                        continue;
                    }
                    //當(dāng)數(shù)據(jù)源第n個(gè)等于根節(jié)點(diǎn)并且已經(jīng)成功添加過(guò)數(shù)據(jù)時(shí)判斷為一段數(shù)據(jù)的結(jié)束,跳出循環(huán),
                    if (s.equals(materialVos.get(count).getHierarchy()) && num) {
                        break;
                    }
                    bomImportVo.add(materialVos.get(count));
                    num = true;
                }
                segmentedData.add(bomImportVo);
            }
        }
        return segmentedData;
    }

手動(dòng)設(shè)置每棵樹(shù)每個(gè)節(jié)點(diǎn)的id以及父id

代碼如下（示例）：

for (List<MatMaterialBomImportVo> segmentedDatum : subsection(materialVos, "0")) {
                //設(shè)置id以及父id
                int i = 0;
                for (MatMaterialBomImportVo vo : segmentedDatum) {
                    BeanTrim.beanAttributeValueTrim(vo);
                    vo.setPrimaryKey(i);
                    getParentId(vo, segmentedDatum);
                    i++;
                }
}
/**
     * 設(shè)置父id
     *
     * @param vo
     * @param segmentedDatum
     */
    private void getParentId(MatMaterialBomImportVo vo, List<MatMaterialBomImportVo> segmentedDatum) {
        for (int j = vo.getPrimaryKey(); j >= 0; j--) {
            if (Integer.parseInt(segmentedDatum.get(j).getHierarchy()) == Integer.parseInt(vo.getHierarchy()) - 1) {
                vo.setForeignKey(segmentedDatum.get(j).getPrimaryKey());
                break;
            }
            if (j == 0) {
                vo.setForeignKey(-1);
            }
        }
    }

說(shuō)明：拆分為若干棵樹(shù)后設(shè)置每條數(shù)據(jù)的虛擬id為自己的索引,每棵樹(shù)的id互相隔離,

根據(jù)表格數(shù)據(jù)規(guī)律得出子節(jié)點(diǎn)只可能存在于自己節(jié)點(diǎn)以下,以及下一個(gè)相同節(jié)地以上,根據(jù)這個(gè)規(guī)律設(shè)置每個(gè)節(jié)點(diǎn)的父id

遞歸封裝為樹(shù)結(jié)構(gòu)

代碼如下（示例）：

/**
     * 遞歸遍歷為樹(shù)形結(jié)構(gòu)
     *
     * @param vo 當(dāng)前處理的元素
     * @param segmentedDatum 每棵樹(shù)的數(shù)據(jù)集
     */
    private void treeData(MatMaterialBomImportVo vo, List<MatMaterialBomImportVo> segmentedDatum) {
        for (int i = vo.getPrimaryKey(); i < segmentedDatum.size(); i++) {
            if (i + 1 == segmentedDatum.size()) {
                if (vo.getForeignKey() == null) {
                    getParentId(vo, segmentedDatum);
                }
                break;
            }
            int v = Integer.parseInt(vo.getHierarchy());
            int vs = Integer.parseInt(segmentedDatum.get(i + 1).getHierarchy());
            if (vs == v + 1) {
                if (v > 1) {
                    vo.setForeignKey(segmentedDatum.get(i).getPrimaryKey());
                    for (int j = vo.getPrimaryKey(); j > 0; j--) {
                        if (Integer.parseInt(segmentedDatum.get(j).getHierarchy()) == Integer.parseInt(vo.getHierarchy()) - 1) {
                            vo.setForeignKey(segmentedDatum.get(j).getPrimaryKey());
                        }
                    }
                }
                vo.getImportVoList().add(segmentedDatum.get(i + 1));
            }
            if (vs <= v) {
                if (vo.getForeignKey() == null) {
                    for (int j = vo.getPrimaryKey(); j > 0; j--) {
                        if (Integer.parseInt(segmentedDatum.get(j).getHierarchy()) == Integer.parseInt(vo.getHierarchy()) - 1) {
                            vo.setForeignKey(segmentedDatum.get(j).getPrimaryKey());
                            break;
                        }
                    }
                }
                break;
            }
        }
        if (vo.getImportVoList() != null && vo.getImportVoList().size() > 0) {
            for (MatMaterialBomImportVo matMaterialBomImportVo : vo.getImportVoList()) {
                treeData(matMaterialBomImportVo, segmentedDatum);
            }
        }
    }

說(shuō)明：我這里傳進(jìn)來(lái)的vo是沒(méi)有設(shè)置id和父id的,只對(duì)數(shù)據(jù)源做了樹(shù)拆分處理,因?yàn)闃I(yè)務(wù)需求,后面并沒(méi)有使用這套遞歸的方法組裝為樹(shù),所以遞歸代碼可能有點(diǎn)誤差,僅供參考