-- --------------------- 閉包表的增刪改查 ----------------------------------------
-- 節點表
DROP TABLE IF EXISTS `tree_node`;
CREATE TABLE `tree_node` (
`id` int(11) UNSIGNED PRIMARY KEY NOT NULL AUTO_INCREMENT COMMENT '主鍵',
`name` varchar(255) DEFAULT NULL COMMENT '節點名稱'
) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='樹節點表';
-- 閉包表
DROP TABLE IF EXISTS `tree_path`;
CREATE TABLE `tree_path` (
`id` int(11) UNSIGNED PRIMARY KEY NOT NULL AUTO_INCREMENT,
`ancestor` int(11) NOT NULL COMMENT '祖先id',
`descendant` int(11) NOT NULL COMMENT '後代id',
`depth` int(11) NOT NULL COMMENT '後代相對祖先的深度'
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='閉包表';
-- 添加 ----
/* 添加如下結構的樹
1
/ \
2 4
/ \ / \
3 5 6 7
/ \
8 9
*/
-- 一、添加節點 ----------------
/*
思路:現在我們要添加一個新的節點3爲節點2的子節點,首先是創建3節點,
之後在關係表中添加三條記錄:
1.其祖先節點和後代節點爲3並且深度爲0的關係記錄(自己和自己的關係)
2.因爲3節點是節點2和1的後代節點,所以我們在關係表中添加兩條記錄:後代1和祖先3深度爲2的關係,後代1和祖先2的深度爲1的關係(自己和祖先的關係)
*/
/*INSERT INTO tree_node(`name`) VALUES('節點1'),('節點2'),('節點3'),('節點4'),('節點5'),('節點6'),('節點7'),('節點8'),('節點9');
-- 添加節點關係 (parent_id,new_id):(1,1)、(1,2)、(2,3)、(1,4)、(2,5)、(4,6)、(4,7)、(6,8)、(6,9)
SET @parent_id = 2; -- 父節點id 1,1,2,1,2,4,4,6,6
SET @new_id = 5; -- 新加節點id 1,2,3,4,5,6,7,8,9
INSERT INTO tree_path(ancestor, descendant, depth)
SELECT t.ancestor, @new_id, t.depth + 1
FROM tree_path AS t WHERE descendant = @parent_id -- 父節點所有祖先(包括自己)
UNION
SELECT @new_id, @new_id, 0; -- 自己是自己的父節點,深度爲0*/
-- 使用存儲過程批量添加
DELIMITER $$
DROP PROCEDURE IF EXISTS `pro_add`;
CREATE PROCEDURE pro_add()
BEGIN
INSERT INTO tree_node(`name`) VALUES('節點1'),('節點2'),('節點3'),('節點4'),('節點5'),('節點6'),('節點7'),('節點8'),('節點9');
SET @parent_ids="1,1,2,1,2,4,4,6,6", @new_ids="1,2,3,4,5,6,7,8,9", @i=1;
SET @count=CHAR_LENGTH(@parent_ids)-CHAR_LENGTH(REPLACE(@parent_ids,',','')) + 1;
WHILE @i <= @count
DO
set @parent_id = SUBSTRING_INDEX(SUBSTRING_INDEX(@parent_ids,',',@i),',',-1);
set @new_id = SUBSTRING_INDEX(SUBSTRING_INDEX(@new_ids,',',@i),',',-1);
-- 依次插入每個成員
INSERT INTO tree_path(ancestor, descendant, depth)
SELECT t.ancestor, @new_id, t.depth + 1
FROM tree_path AS t WHERE descendant = @parent_id -- 父節點所有祖先(包括自己)
UNION
SELECT @new_id, @new_id, 0; -- 自己是自己的父節點,深度爲0
SET @i=@i+1;
END WHILE;
END$$
DELIMITER ;
CALL pro_add();
-- 二、查詢節點 ---------------- 根據關係表查詢即可
-- 查詢後代節點(包括自己)
SELECT descendant
FROM tree_path
WHERE ancestor = 6
ORDER BY depth ASC;
-- 查詢所有祖先節點(包括自己)
SELECT ancestor
FROM tree_path
WHERE descendant = 7
ORDER BY depth DESC;
-- 三、刪除節點 ----------------
/**
假如我們需要刪除2節點,也就是連同2節點的後代全部刪除,
先找到2節點的所有後代集,在節點表中把他們刪掉,
之後需要刪除節點2所有後代節點(包括自己)與祖先的關係
*/
-- 刪除指定節點
SET @del_id = 2; -- 要刪除的id
-- 先其後代節點(包括自己)
DELETE FROM tree_node WHERE id IN (SELECT descendant FROM tree_path WHERE ancestor = @del_id);
-- 再刪除其所有後代(包括自己)節點與祖先的關係
DELETE FROM tree_path WHERE descendant IN(SELECT descendant FROM (SELECT descendant, ancestor FROM tree_path) t WHERE ancestor = @del_id);
-- 四、移動子樹 ----------------
/**
將節點6 移到 節點1節點下面。
第一步需要刪除 6節點的祖先節點集(不包括自己:4和1)和6節點的後代節點集(包括自己:6、8和9)的關係。
第二步重新建立關係,把(1節點的祖先節點集(包括1))X(6節點的後代節點集(包括6))的笛卡爾集插入到關係表中,
其深度爲第一個集合的深度加上第二個集合的深度再加一。
*/
-- 1.分離子樹,刪除節點6的祖先和後代節點的關係
SET @from_id = 6;
SET @to_id = 1;
DELETE FROM tree_path
WHERE
descendant IN (SELECT d.descendant FROM (SELECT * FROM tree_path) d WHERE d.ancestor=@from_id) -- 後代節點(包括自己)
AND
ancestor IN (SELECT a.ancestor FROM (SELECT * FROM tree_path) a WHERE a.descendant=@from_id AND a.ancestor<>a.descendant); -- 祖先節點
-- 2.將上一步分離出的子樹用笛卡爾積嫁接到節點1下
INSERT INTO tree_path(ancestor,descendant,depth)
SELECT
super.ancestor,
sub.descendant,
super.depth + sub.depth + 1
FROM
tree_path super
CROSS JOIN
tree_path sub
WHERE
super.descendant=@to_id AND sub.ancestor=@from_id;
-- 五、其它查詢
-- 查詢根
SELECT ancestor
FROM tree_path
WHERE descendant = 7
ORDER BY depth DESC
LIMIT 1;
-- 查詢子節點(一級後代)
SELECT descendant
FROM tree_path
WHERE ancestor = 4 AND depth = 1;
-- 查詢父節點(一級祖先)
SELECT ancestor
FROM tree_path
WHERE descendant = 4 AND depth = 1;
-- 查詢所有葉子節點與其祖父的關係鏈
SELECT descendant, GROUP_CONCAT(ancestor ORDER BY depth DESC SEPARATOR ',')
FROM tree_path
-- where descendant = 5
group by descendant;
推薦看圖理解連接:閉包表圖解