设计一个A表数据抽取到B表的抽取过程

原题如下：

解题代码如下：

table1类：

@Data
@NoArgsConstructor
@AllArgsConstructor
public class table1{
    private String num;
    private String name;
    private String fatherNum;
}

table2类：

@Data
@NoArgsConstructor
@AllArgsConstructor
public class table2{
    private String num;
    private String name;
    private String sheng;
    private String shi;
    private String qu;
}

changeTable类：

@Data
@NoArgsConstructor
@AllArgsConstructor
class Node{
    //编号
    private String num;
    //名称
    private String name;
    //父节点
    private Node fatherNum;
    //子节点树
    private List<Node> sonNum;
}
@Data
public class changeTable {
    //零散节点，key为编号
    private Map<String,Node> MapData = new HashMap<>();
    //结构同HashMap，list中存储的每个节点其形状都为树形，其中根为省节点
    private List<Node> treeData = new ArrayList<>();
    //处理完后生成的新表
    private List<table2> tab = new ArrayList<>();
    private void createTree(List<table1> datas) {
        for(table1 data : datas) {
            //为省列，直接添加到treeData中，形成树根
            if(data.getFatherNum()==null) {
                treeData.add(MapData.get(data.getNum()));
            }else {
                //当前节点中设置父亲点
                MapData.get(data.getNum()).setFatherNum(MapData.get(data.getFatherNum()));
                //父节点中设置孩子节点
                List<Node> sonNum = MapData.get(data.getFatherNum()).getSonNum();
                sonNum.add(MapData.get(data.getNum()));
                MapData.get(data.getFatherNum()).setSonNum(sonNum); 
            }
        }
    }
    //生成零散节点，方便获取
    private void createMap(List<table1> datas) {
        for(table1 data:datas) {
            MapData.put(data.getNum(), new Node(data.getNum(),data.getName(),new Node(),new ArrayList<Node>()));
        }
    }
    //生成新表数据
    private void createTable2() {
        for(Node node : treeData) {
            //node为树根，0为深度
            createRow(node,0);
        }
    }
    //使用深度优先遍历产生数据
    private void createRow(Node node,int depth) {
        //当能达到第三层（0层：省，1层：市，2层：区，3层：公司名），则新增数据
        if(depth==3) {
            table2 table2 = new table2();
            table2.setNum(node.getNum());
            table2.setName(node.getName());
            table2.setQu(node.getFatherNum().getNum());
            table2.setShi(node.getFatherNum().getFatherNum().getNum());
            table2.setSheng(node.getFatherNum().getFatherNum().getFatherNum().getNum());
            tab.add(table2);
        }else {
            //遍历当前节点子节点树
            for(Node nod : node.getSonNum()) {
                createRow(nod,depth+1);
            }
        }
    }
    public List<table2> getTable2(List<table1> dataList){
        createMap(dataList);
        createTree(dataList);
        createTable2();
        return tab;
    }
}

Test类：

public class Test {
    public static void main(String[] args) {
        //模拟查询数据库
        List<table1> dataList = new ArrayList<>();
        dataList.add(new table1("345872234865","江汉区XXX公司","342309876534"));
        dataList.add(new table1("348724235332","长沙市","348653423233"));
        dataList.add(new table1("348765064386","武汉市","348653458722"));
        dataList.add(new table1("348652342344","岳麓区","348724235332"));
        dataList.add(new table1("348653458722","湖北省",null));
        dataList.add(new table1("425232324523","岳麓区ZZZ公司","348652342344"));
        dataList.add(new table1("348653423233","湖南省",null));
        dataList.add(new table1("342309876534","江汉区","348765064386"));
        dataList.add(new table1("5065438634876","江汉区YYY公司","342309876534"));
        //创建转换对象
        changeTable changeTable = new changeTable();
        //得到转换结果
        List<table2> table2s = changeTable.getTable2(dataList);
        for(table2 tab :table2s ) {
            System.out.println(tab);
        }
    }
}

算法解析：

注释已经解释的非常详细了，具体过程我就不再叙述了。

其中table1与table2是两个po对象，而@data注解相当于被注解类中所有成员加上了get、set方法，@NoArgsConstructor为无参构造方法，@AllArgsConstructor为全参构造方法。

运行结果：

注意：当数据非常多时，这种设计将不再适用，因为将所有数据加载到内存中会导致栈溢出。这时可以先将数据库中所有省列查询出来，(省列没有上级编号，这就是很好的查询条件)；接着循环遍历省列，根据遍历的省查询出下面所有市；然后遍历查询出来的所有市得到市下面的所有区；然后遍历区下面的所有公司，最后组装数据。虽然性能远不如一次性导入数据，但针对于数据量大时是非常好的解决办法。