JAVA代码实现多级树结构封装对象

我们经常在代码里会造一个树结构对象，以方便前端使用。

以地区（区、镇、村）为例
后台一般对于树结构对象在数据库的结构是这样的：

主键ID 名字 父ID ID REGION_NAME PARENT_ID 121100 尼龙区 0 121100001 尼龙区钢丝镇 121100 121100001001 尼龙区钢丝镇螺丝村 121100001

  其实这样返回给前端，前端是可以做成树的，但是我这里不是要说这种的，我遇到的是下边的情况。

主键ID 名字 词典表类型 ID REGION_NAME DICT_TYPE 121100 尼龙区 XZQH 121100001 尼龙区钢丝镇 XZQH 121100001001 尼龙区钢丝镇螺丝村 XZQH 231111 一般性支付 ZFFS

  数据库的结构如上，并没有树结构，需要根据DICT_TYPE为XZQH的记录取出来，然后封装成树结构返回给前端，前端要的树结构类似如下：

我的实现的逻辑是这样的：

• 首先根据DICT_TYPE取出相关的地区记录。
• 然后将区、镇、村分别存到三个不同的map里。
• 最后，嵌套遍历三个map的key，区里边套镇、镇里边套村，封装对象。（最重要的地方）。

具体步骤：

1、建立一个对象，用来规范树结构。

public class FpzjXzdwArray implements Serializable  {    private String value; //对应数据库的ID    private String text; //对应数据库的region_name    private List<FpzjXzdwArray> children;    //省略get/set方法了}

2、第一步我就省略了。

3、第二步像我这种情况就是根据ID了，长度为6的放到区map里，长度为9的放到镇map里，长度为12的放到村map里。

Map<String,FpzjXzdwArray> quMap = new HashMap<String,FpzjXzdwArray>();Map<String,FpzjXzdwArray> zhenMap = new HashMap<String,FpzjXzdwArray>();Map<String,FpzjXzdwArray> cunMap = new HashMap<String,FpzjXzdwArray>();

4、嵌套遍历三个map,首先缕缕我们的步骤。

  以一个小例子为例，我们现在有一个大箱子、2个中箱子（1红、1蓝）、6个红蓝小箱子（3红、3蓝）。要求是把大箱子里装上中箱子，中箱子里要颜色对应装进去小箱子。
 正确逻辑：如果让一个正常人来做，我们的逻辑都是一样的，要先把颜色对应的小箱子放到中箱子里以后，才会把中箱子放到大箱子里，对吗？
 错误逻辑：如果你非要先把中箱子放到大箱子以后，再去拿对应颜色的小箱子放进中箱子里去的话，这样带来的一个问题就是，你必须要先把大箱子打开才可以完成这种事。所以这种逻辑就是错误的。

  这个例子反映到代码里也是一样，如果你先往区对象里塞进镇对象集合，然后再去往镇对象里塞对应的村对象，就会像上边的第二个步骤一样，逻辑就错了。我的代码如下：（最下边还有别人分享给我的好方法，先贴上我的，瞬间感觉自己弱爆了）

for(String qu:quMap.keySet()){ //遍历所有的区            for(String zhen:zhenMap.keySet()){ //遍历所有的镇                if(qu.equals(zhen.substring(0,6))){ //该区下找到该镇                    FpzjXzdwArray zhenXzdw = null;//定义这个是有必要的，这样可以先完成对村对象塞到镇对象里，然后再完成镇对象塞到区对象的步骤。                    for(String cun:cunMap.keySet()){ //有区有镇，然后再找下一级村                        if(zhen.equals(cun.substring(0,9))){ //该镇下找到村                            zhenXzdw = zhenMap.get(zhen); //得到该镇对象，                            List<FpzjXzdwArray> xzdwArrayZhenList = zhenXzdw.getChildren(); //取出镇对象下的村集合                            if(xzdwArrayZhenList == null){                                xzdwArrayZhenList = new ArrayList<>();                                xzdwArrayZhenList.add(cunMap.get(cun));                                zhenXzdw.setChildren(xzdwArrayZhenList); //将符合该镇下的村塞到镇对象里                            }else{                                xzdwArrayZhenList.add(cunMap.get(cun));                                zhenXzdw.setChildren(xzdwArrayZhenList); //将符合该镇下的村塞到镇对象里                            }                        }                    }                    FpzjXzdwArray quXzdw = quMap.get(qu); //得到该区对象，                    List<FpzjXzdwArray> xzdwArrayList = quXzdw.getChildren();//取出区对象下的镇集合                    if(xzdwArrayList == null){                        xzdwArrayList = new ArrayList<>();                        xzdwArrayList.add(zhenXzdw);                        quXzdw.setChildren(xzdwArrayList); //将符合该区下的镇对象塞到区对象里                        quMap.put(qu,quXzdw);                    }else{                        xzdwArrayList.add(zhenXzdw);                        quXzdw.setChildren(xzdwArrayList);//将符合该区下的镇对象塞到区对象里                        quMap.put(qu,quXzdw);                    }                }            }        }        return quMap.values();

以上就是我写的实现的逻辑，我还没有用上递归，本质是这样。
消耗的时间我也测试了，对我来说几毫秒，最多7ms。
我的数据的数量是：

数量 区 镇 村 总数 8 66 674

遍历的总次数：8*66*674=？ //其实也挺多的次数。。

分享一：

  在我加的技术群里吆喝了半天，有一位朋友分享他们的工具类，我感觉写的很好了。他需要数据的结构是：

主键ID 名字 父ID ID REGION_NAME PARENT_ID 121100 尼龙区 0 121100001 尼龙区钢丝镇 121100 121100001001 尼龙区钢丝镇螺丝村 121100001

对应封装好的实体类对象可以是这样的：

public class TestTree{    private String id;    private String name;    private String pid;    private List<TestTree> testTrees;    //省略get/set方法 }

赋上他给我的两个工具类：
工具类下载地址
使用的主要逻辑在TreeUtils类里，在这里边我加了一个main方法，模拟了一个例子，大家可以参考。最后实现的结果如下图：

他的这个工具类的好处是，利用了反射，只要你的数据结构符合上边的id、pid、name这样的数据，叫什么名字都无所谓，不管多少层，都可以给弄出来。觉得这个特别好了。感谢@小太阳の ！！

造树我觉得是挺重要的，挺常用的，如果有多级树，我这种方式就不适用，所以我想找到一个可以封装多级树的好代码，希望有能看到，留言共享一下，我再把好的方法补充到这个文章里。