MySQL 查询树结构方式

mysql 查询树结构

1. 关于树结构

此类结构的数据，通常需要表结构中含有id 、parentid等自关联字段，有时为了提高查询效率还可增加更多冗余字段，如index，index的值为所有父级目录的id字符串集合。

关于树结构数据的组装，常见的写法是在程序中通过递归的方式去构建出一颗完整的树，单纯通过sql的方式其实并不常用，下面分别给出两种方式的例子。

2. mysql自定义函数的方式

什么是mysql自定义函数：聚合函数，日期函数之类的都是mysql的函数，此处我们定义的函数可同他们一样使用，不过只能在定义的数据库中使用，自定义函数和存储过程类似，不同的是，函数只会返回一个值，不允许返回一个结果集。

2.1 创建测试数据

create table `tree`  (
  `id` bigint(11) not null,
  `pid` bigint(11) null default null,
  `name` varchar(255) character set utf8 collate utf8_general_ci null default null,
  primary key (`id`) using btree
) engine = innodb character set = utf8 collate = utf8_general_ci row_format = dynamic;
insert into `tree` values (1, 0, '中国');
insert into `tree` values (2, 1, '四川省');
insert into `tree` values (3, 2, '成都市');
insert into `tree` values (4, 3, '武侯区');
insert into `tree` values (5, 4, '红牌楼');
insert into `tree` values (6, 1, '广东省');
insert into `tree` values (7, 1, '浙江省');
insert into `tree` values (8, 6, '广州市');

2.2 获取 某节点下所有子节点

create function `get_child_node`(rootid varchar(100))   
returns varchar(2000)  
begin   
declare str varchar(2000);  
declare cid varchar(100);   
set str = '$';   
set cid = rootid;   
while cid is not null do   
    set str = concat(str, ',', cid);   
    select group_concat(id) into cid from tree where find_in_set(pid, cid);   
end while;   
return str;   
end

调用自定义函数

select * from tree where find_in_set(id, get_child_node(2));

2.3 获取 某节点的所有父节点

create function `get_parent_node`(rootid varchar(100))   
returns varchar(1000)   
begin   
declare fid varchar(100) default '';   
declare str varchar(1000) default rootid;   
  
while rootid is not null do   
    set fid =(select pid from tree where id = rootid);   
    if fid is not null then   
        set str = concat(str, ',', fid);   
        set rootid = fid;   
    else   
        set rootid = fid;   
    end if;   
end while;   
return str;  
end

调用自定义函数

select * from tree where find_in_set(id, get_parent_node(5));

3. oracle数据库的方式

只需要使用start with connect by prior语句即可完成递归的树查询，详情请自己查阅相关资料。

4. 程序代码递归的方式构建树

这里我就不给出完整代码了，递归的方式很简单，就是先查出所有树节点，然后通过一个treenode类中的add方法递归把所有子节点给加进来。核心代码如下：

public class treenodedto {
    
    private string id;
    private string parentid;
    private string name;
    private list<treenodedto> children = new arraylist<>();
    public void add(treenodedto node) {
        if ("0".equals(node.parentid)) {
            this.children.add(node);
        } else if (node.parentid.equals(this.id)) {
            this.children.add(node);
        } else {
         	//递归调用add()添加子节点
            for (treenodedto tmp_node : children) {
                tmp_node.add(node);
            }
        }
    }
 }

5. 通过hashmap，只需要遍历一次

就可以完成树的生成：五星推荐

list<treenodedto> list = dbmapper.getnodelist();
arraylist<treenodedto> rootnodes = new arraylist<>();
map<integer, treenodedto> map = new hashmap<>();
for (treenodedto node :list) {
    map.put(node.getid(), node);
    integer parentid = node.getparentid();
    // 判断是否有父节点 (没有父节点本身就是个父菜单)
    if (parentid.equals('0')){
        rootnodes.add(node);
        // 找出不是父级菜单的且集合中包括其父菜单id
    } else if (map.containskey(parentid)){
        map.get(parentid).getchildren().add(node);
    }
}

mysql 查询带树状结构的信息

在oracle中有函数应用直接能够查询出树状的树状结构信息，例如有下面树状结构的组织成员架构，那么如果我们想查其中一个节点下的所有节点信息

在oracle中可以直接用下面的语法可以进行直接查询

start with connect by prior

但是在mysql中是没有这个语法的

而如果你也是想要查询这样的数据结构信息该怎么做呢？我们可以自定义函数。我们将上面的信息初始化信息进数据库中。首先先创建一张表用于存储这些信息，id为存储自身的id信息，parent_id存储父id信息

create table `company_inf` (
  `id` varchar(32) collate utf8mb4_unicode_ci default null,
  `name` varchar(255) collate utf8mb4_unicode_ci default null,
  `parent_id` varchar(32) collate utf8mb4_unicode_ci default null
)

然后将图中的信息初始化表中

insert into company_inf values ('1','总经理王大麻子','1');
insert into company_inf values ('2','研发部经理刘大瘸子','1');
insert into company_inf values ('3','销售部经理马二愣子','1');
insert into company_inf values ('4','财务部经理赵三驼子','1');
insert into company_inf values ('5','秘书员工j','1');
insert into company_inf values ('6','研发一组组长吴大棒槌','2');
insert into company_inf values ('7','研发二组组长郑老六','2');
insert into company_inf values ('8','销售人员g','3');
insert into company_inf values ('9','销售人员h','3');
insert into company_inf values ('10','财务人员i','4');
insert into company_inf values ('11','开发人员a','6');
insert into company_inf values ('12','开发人员b','6');
insert into company_inf values ('13','开发人员c','6');
insert into company_inf values ('14','开发人员d','7');
insert into company_inf values ('15','开发人员e','7');
insert into company_inf values ('16','开发人员f','7');

例如我们想要查询研发部门经理刘大瘸子下的所有员工，在oracle中我们可以这样写

  select *
  from t_portal_authority
  start with id='1'
  connect by prior id = parent_id

而在mysql中我们需要下面这样自定义函数

create function getchild(parentid varchar(1000))
returns varchar(1000)
begin
    declare otemp varchar(1000);
    declare otempchild varchar(1000);
    set otemp = '';
    set otempchild =parentid;
    while otempchild is not null do
        if otemp != '' then
            set otemp = concat(otemp,',',otempchild);
        else
            set otemp = otempchild;
        end if;
        select group_concat(id) into otempchild from company_inf where parentid<>id and find_in_set(parent_id,otempchild)>0;
    end while;
return otemp;
end

然后这样查询即可

select * from company_inf where find_in_set(id,getchild('2'));

此时查看查询出来的信息就是刘大瘸子下所有的员工信息了

以上为个人经验，希望能给大家一个参考，也希望大家多多支持。