Mysql 閉包表的增刪改查實現

-- --------------------- 閉包表的增刪改查  ----------------------------------------
-- 節點表
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;

推薦看圖理解連接:閉包表圖解

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章