Skip to content

Instantly share code, notes, and snippets.

@alikon
Created September 14, 2019 10:14
Show Gist options
  • Select an option

  • Save alikon/fc3bdaadcd999a29c9b2b97c00e3214e to your computer and use it in GitHub Desktop.

Select an option

Save alikon/fc3bdaadcd999a29c9b2b97c00e3214e to your computer and use it in GitHub Desktop.
DROP PROCEDURE IF EXISTS tree_recover;
DELIMITER //
CREATE PROCEDURE tree_recover ()
MODIFIES SQL DATA
BEGIN
DECLARE currentId, currentParentId INT;
DECLARE currentLeft INT;
DECLARE startId INT DEFAULT 0;
# Determines the max size for MEMORY tables.
SET max_heap_table_size = 1024 * 1024 * 512;
# START TRANSACTION;
DROP TABLE `tmp_tree`;
# Temporary MEMORY table to do all the heavy lifting in,
# otherwise performance is simply abysmal.
CREATE TABLE `tmp_tree` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`parent_id` int(10) unsigned NOT NULL DEFAULT 0 ,
`lft` int(11) unsigned DEFAULT NULL,
`rgt` int(11) unsigned DEFAULT NULL,
PRIMARY KEY (`id`),
INDEX (`parent_id`),
INDEX (`lft`),
INDEX (`rgt`)
) ENGINE = MEMORY
SELECT `id`,
`parent_id`,
`lft`,
`rgt`
FROM `my4_menu`;
# Leveling the playing field.
UPDATE `tmp_tree`
SET `lft` = NULL,
`rgt` = NULL;
# Establishing starting numbers for all root elements.
WHILE EXISTS (SELECT * FROM `tmp_tree` WHERE `parent_id` =0 AND `lft` IS NULL AND `rgt` IS NULL LIMIT 1) DO
UPDATE `tmp_tree`
SET `lft` = startId,
`rgt` = startId + 1
WHERE `parent_id` =0
AND `lft` IS NULL
AND `rgt` IS NULL
LIMIT 1;
SET startId = startId + 2;
END WHILE;
# Switching the indexes for the lft/rgt columns to B-Trees to speed up the next section, which uses range queries.
DROP INDEX `lft` ON `tmp_tree`;
DROP INDEX `rgt` ON `tmp_tree`;
CREATE INDEX `lft` USING BTREE ON `tmp_tree` (`lft`);
CREATE INDEX `rgt` USING BTREE ON `tmp_tree` (`rgt`);
# Numbering all child elements
WHILE EXISTS (SELECT * FROM `tmp_tree` WHERE `lft` IS NULL LIMIT 1) DO
# Picking an unprocessed element which has a processed parent.
SELECT `tmp_tree`.`id`
INTO currentId
FROM `tmp_tree`
INNER JOIN `tmp_tree` AS `parents`
ON `tmp_tree`.`parent_id` = `parents`.`id`
WHERE `tmp_tree`.`lft` IS NULL
AND `parents`.`lft` IS NOT NULL
LIMIT 1;
# Finding the element's parent.
SELECT `parent_id`
INTO currentParentId
FROM `tmp_tree`
WHERE `id` = currentId;
# Finding the parent's lft value.
SELECT `lft`
INTO currentLeft
FROM `tmp_tree`
WHERE `id` = currentParentId;
# Shifting all elements to the right of the current element 2 to the right.
UPDATE `tmp_tree`
SET `rgt` = `rgt` + 2
WHERE `rgt` > currentLeft;
UPDATE `tmp_tree`
SET `lft` = `lft` + 2
WHERE `lft` > currentLeft;
# Setting lft and rgt values for current element.
UPDATE `tmp_tree`
SET `lft` = currentLeft + 1,
`rgt` = currentLeft + 2
WHERE `id` = currentId;
END WHILE;
# Writing calculated values back to physical table.
UPDATE `my4_menu`, `tmp_tree`
SET `my4_menu`.`lft` = `tmp_tree`.`lft`,
`my4_menu`.`rgt` = `tmp_tree`.`rgt`
WHERE `my4_menu`.`id` = `tmp_tree`.`id`;
COMMIT;
DROP TABLE `tmp_tree`;
END//
DELIMITER ;
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment