树形结构与SQL的碰撞——三种树形结构存储方案的对比

我们在业务中经常要处理树形结构的数据,比如目录树、组织关系、血缘关系等。那么在关系型数据库中如何高效维护并查询树形结构呢? 本文将介绍并对比常见的三种树形结构的存储方案,供大家选择

0. 数据准备

为了方便下文的描述,本文引用《SQL Antipatterns》中评论的例子(本质也是一棵树的结构)进行对比,其树形结构如下:

image-20211207204927731

1. 路径树 Path Enum

路径树,也被成为前缀树,即用从根节点到当前节点的路径来表示节点在树中的位置,其表结构为:

CREATE TABLE path_block_tree (
  block_id   SERIAL PRIMARY KEY,
  path         VARCHAR(100000),  -- 树路径
  parent_id    BIGINT UNSIGNED,
  FOREIGN KEY (parent_id) REFERENCES adjacency_block_tree(block_id)
);

表中内容为:


image-20211207203708990

其中path字段就是树路径,从根节点到当节点路径树上每一个节点ID都用"/"连接在一起。

查询

如果要查询某个节点向上的根路径或者所有的子节点,可以直接根据path字段进行拼配,SQL简单高效。当然为了提高查询效率,path字段需要加索引。

-- 根路径 向上
SELECT *
FROM path_block_tree AS c
WHERE '1/4/6/7/' LIKE concat(c.path,'%');

-- 子树 向下
SELECT *
FROM path_block_tree AS c
WHERE c.path LIKE concat('1/4/','%');

树结构维护

  1. 插入:根据父节点path信息,加入自身节点ID,插入子节点;
  2. 删除:查询删除节点的子树,然后批量删除;
  3. 修改:查询修改节点子树,然后根据修改节点更新后的path,批量替换子节点的path;

2. 闭包表 Closure Table

闭包表就是用闭包结构来表示树形结构的表,换句话就是表中包含全部节点是的前驱后继关系,在路径表的基础上,还需要添加如下表结构:

CREATE TABLE closureTree (
  ancestor    BIGINT UNSIGNED NOT NULL,
  descendant  BIGINT UNSIGNED NOT NULL,
  depth BIGINT UNSIGNED NOT NULL,
  PRIMARY KEY(ancestor, descendant),
  FOREIGN KEY (ancestor) REFERENCES  path_block_tree(block_id),
  FOREIGN KEY (descendant) REFERENCES path_block_tree(block_id)
);

表中数据如图:

闭包表中数据

表中闭包的关联:


image-20211207205048752

查询

对于闭包表,可以通过表之间的Join来实现树形结构的查询。

-- 查询节点4的子树
SELECT c.*
FROM path_block_tree AS c
  JOIN closureTree AS t ON c.block_id = t.descendant
WHERE t.ancestor = 4;

--  查询节点6的根路径
SELECT c.*
FROM path_block_tree AS c
  JOIN closureTree AS t ON c.block_id = t.ancestor
WHERE t.descendant = 6;、

树结构维护

  1. 删除节点:先查询子树,然后批量删除

  2. 插入节点:比如将节点8 插入到节点5下面,需要批量插入如下数据

    INSERT INTO closure_tree_path (ancestor, descendant, depth)
      SELECT ancestor, '{$node_id}', depth+1 FROM closure_tree_path
      WHERE descendant = '{$parent_id}'
      UNION ALL SELECT '{$node_id}', '{$node_id}', 0;
    
  3. 修改节点:操作比较困难,需要先批量删除所有闭包表中相关节点数据,然后重新依次插入;

3. 邻接表 Adjacency Tree

邻接表是最简单通用也是最好理解的树的存储方式,其表机构如下:

表结构

CREATE TABLE adjacency_block_tree (
  block_id     SERIAL PRIMARY KEY, # an alias for BIGINT UNSIGNED NOT NULL AUTO_INCREMENT UNIQUE
  parent_id    BIGINT UNSIGNED,
  FOREIGN KEY (parent_id) REFERENCES adjacency_block_tree(block_id)
);

邻接表存储数据很简单,但是要查询就比较麻烦,需要在代码里递归查询每一层数据,这样比较耗时, 幸好Mysql 8已经支持了递归查询。

递归查询并不是什么新鲜事物,其实早在 SQL 99标准中已经规定了通过 WITH 来定义通用的表的定义表达式,Microsoft SQL Server 2005, Oracle 11g, IBM DB2, PostgreSQL 8.4 等数据库都支持递归表达式,而MySQL 8.0时代也支持了该功能(虽然晚了好多年)。

查询子树

-- 子树查询(向下)
WITH RECURSIVE blockTree(block_id, parent_id, depth)
      AS (
          SELECT *, 0 AS depth FROM adjacency_block_tree
          WHERE block_id = ?
        UNION ALL
          SELECT c.*, ct.depth +1 AS depth FROM blockTree ct
          JOIN adjacency_block_tree c ON (ct.block_id = c.parent_id)
      )
SELECT * FROM blockTree order by depth, block_id;

查询树路径

-- 路径查询(向上)
WITH RECURSIVE blockTree(block_id, parent_id, depth)
      AS (
          SELECT *, ? AS depth FROM adjacency_block_tree
          WHERE block_id = ?
        UNION ALL
          SELECT c.*, ct.depth - 1 AS depth FROM blockTree ct
          JOIN adjacency_block_tree c ON (ct.parent_id = c.block_id)
      )
SELECT * FROM blockTree order by depth, block_id;

查询效率

那这样递归查DB的性能如何呢?会不会给mysql造成压力,笔者在2C8G的VM进行压测,分别执行根节点的子树查询,其性能表现如下:

节点个数 树的最大深度 执行时间
10 3-depth 44.5ms
100 10-depth 48ms
1000 15-depth 51ms
10000 20-depth 137ms
100000 30-depth 309ms

可见其性能在1W节点以内的树应用场景是可以满足的。

数据维护

邻接表的优势在于维护树结构简单:

  • 插入节点:单SQL插入一行数据;

  • 修改节点:单SQL修改一行数据;

  • 删除节点:先查询出该节点子树的全部节点,然后在批量删除;

4. 总结

笔者用一张表来对比下三种方案的特性。

方案 储存 数据导出 查询父子关系 查询子树 查询根路径 插入 删除 修改 并发 备注
连接表&迭代查询 较少 Easy Easy 迭代查询 Easy 迭代查询Easy Easy 先查询子节点再批量删除 Easy 容易控制 必须依赖Mysql 8.0
路径树 path可能很长 Easy Easy Easy Easy Easy Easy Esay 路径上锁 Path过长,效率不高
闭包表 较多 Hard Easy Easy Easy a little hard Easy Hard Hard 支持多父节点

没有最完美的方案,只有最适合。如果读者还在犹豫不决,笔者可以根据自己的经验来做个简单直接的推荐

  1. 如果树的结构变化频率不大,且深度不大,推荐路径树,反之ID为很长的UUID,则不推荐;
  2. 如果需要支持多个父节点的情况,支持无限深度,且不用考虑高并发的话,推荐闭包表;
  3. 如果DB支持递归查询,则建议直接考虑邻接表+递归查询的方式;

你可能感兴趣的:(树形结构与SQL的碰撞——三种树形结构存储方案的对比)