MySQL存储过程中的父/子/孙层次

Question

复制记录我有3个表的父，子&孙子的形式为：

+----------------+ +----------------+ +---------------------+ 
| parent   | | child   | | grandchild   | 
+----------------+ +----------------+ +---------------------+ 
| parent_id (PK) | | child_id (PK) | | grandchild_id (PK) | 
| parent_data | | child_data  | | grandchild_data  | 
+----------------+ | parent_id (FK) | | child_id (FK)  | 
        +----------------+ +---------------------+ 

PK =自动递增的主键。
FK =外键。

我想要一个可以复制父表中的记录和子表和孙表中的任何关联记录的存储过程。我可以将父数据和子数据复制好，它是我正在努力的孙表。这是我得到的：

CREATE FUNCTION sf_copy_parent(p_parent_id INT) RETURNS INT 
BEGIN 
    DECLARE new_parent_id INT; 

    -- create new parent record 
    INSERT INTO parent(parent_data) 
     SELECT parent_data FROM parent 
     WHERE parent_id=p_parent_id; 
    SET new_parent_id=LAST_INSERT_ID(); 

    -- copy child records 
    INSERT INTO child(child_data,parent_id) 
     SELECT child_data,new_parent_id FROM child 
     WHERE parent_id=p_parent_id; 

    -- copy grandchild records ??? 


    -- return 
    RETURN new_parent_id; 
END 

如果这很重要，我正在使用Mysql5.5。

Answer 1

试试这个SELECT查询（它使用“p_parent_id”和“new_parent_id”变量） -

SET @r1 = 1; 
SET @child_id = NULL; 
SET @r2 = 0; 

SELECT c1.grandchild_data, c2.child_id FROM (
    SELECT @r1 := if(c.child_id IS NULL OR c.child_id <> @child_id, @r1 + 1, @r1) rank, @child_id := c.child_id, c.child_id, g.grandchild_data FROM child c 
    JOIN grandchild g 
    ON c.child_id = g.child_id 
    WHERE 
    c.parent_id = p_parent_id 
    ORDER BY 
    c.child_id, g.grandchild_id 
) c1 
JOIN (SELECT @r2 := @r2 + 1 rank, child_id FROM child WHERE parent_id = new_parent_id ORDER BY child_id) c2 
    ON c1.rank = c2.rank; 

如果一切正常，我们将它改写为INSERT..SELECT声明，或尝试自己做; ）

Answer 2

编辑：我从那以后创造了另一个答案，我认为它比这个更简单，更好。

完成的存储过程，然后使用回答@Devart是：

CREATE FUNCTION `sp_copy`(p_parent_id INT) RETURNS int(11) 
BEGIN 
    DECLARE new_parent_id INT; 

    -- create new parent record 
    INSERT INTO parent(parent_data) 
     SELECT parent_data FROM parent 
     WHERE parent_id=p_parent_id; 
    SET new_parent_id=LAST_INSERT_ID(); 

    -- copy child records 
    INSERT INTO child(child_data,parent_id) 
     SELECT child_data,new_parent_id FROM child 
     WHERE parent_id=p_parent_id; 

    -- copy grandchild records 
    SET @r1 = 1; 
    SET @child_id = NULL; 
    SET @r2 = 0; 
    INSERT INTO grandchild(grandchild_data,child_id) SELECT c1.grandchild_data, c2.child_id FROM (
    SELECT @r1 := if(c.child_id IS NULL OR c.child_id <> @child_id, @r1 + 1, @r1) rank, @child_id := c.child_id, c.child_id, g.grandchild_data FROM child c 
    JOIN grandchild g 
     ON c.child_id = g.child_id 
    WHERE 
     c.parent_id = p_parent_id 
    ORDER BY 
     c.child_id, g.grandchild_id 
    ) c1 
    JOIN (SELECT @r2 := @r2 + 1 rank, child_id FROM child WHERE parent_id = new_parent_id ORDER BY child_id) c2 
     ON c1.rank = c2.rank; 

    -- return new parent id 
    RETURN new_parent_id; 
END 

Answer 3

我自从学会了一点，我相信从@Devart的贡献可以简化一点。我现在使用的解决方案如下所示。

注意：如果使用的表是InnoDB而不是MyISAM，此解决方案才能正常工作。我认为这与MyISAM的默认排序行为的工作方式有关。 （InnoDB使用主键，而MyISAM使用插入顺序。）特别是，使用MyISAM表时，'副本孙'（s）'部分生成的两组等级不相互'同步'。在相关部分中添加SORT BY子句似乎也没有区别。

CREATE PROCEDURE sp_copy(p_parent_id INT) 
BEGIN 
    DECLARE new_parent_id INT; 

    -- copy parent 
    INSERT INTO parent(parent_data) SELECT parent_data FROM parent WHERE parent_id=p_parent_id; 
    SET new_parent_id:=LAST_INSERT_ID(); 

    -- copy child(s) 
    INSERT INTO child(child_data, parent_id) 
     SELECT child_data, new_parent_id FROM child WHERE parent_id=p_parent_id; 

    -- copy grandchild(s) 
    SET @rank1:=0;   
    SET @rank2:=0;    
    INSERT INTO grandchild(grandchild_data, child_id) SELECT gc.grandchild_data, c2.child_id FROM 
     (SELECT child_id, @rank1:=@rank1+1 as rank FROM child WHERE parent_id=p_parent_id) c1 
     INNER JOIN 
     (SELECT child_id, @rank2:=@rank2+1 as rank FROM child WHERE parent_id=new_parent_id) c2 ON c1.rank=c2.rank 
     INNER JOIN grandchild gc ON c1.child_id=gc.child_id; 

END 

此外，要处理一个伟大的孩子表格，可以使用复制孙孩子记录时的相同原理。唯一的额外复杂性是在两个子查询的每一个中添加一个连接。这是必需的，因为child.parent_id字段需要在WHERE子句中引用：

-- copy greatgrandchild(s) 
SET @rank1:=0;   
SET @rank2:=0;    
INSERT INTO greatgrandchild(greatgrandchild_data, grandchild_id) SELECT ggc.greatgrandchild_data, gc2.grandchild_id FROM 
    (SELECT grandchild_id, @rank1:=@rank1+1 as rank FROM grandchild INNER JOIN child ON child.child_id=grandchild.child_id WHERE parent_id=p_parent_id) gc1 
    INNER JOIN 
    (SELECT grandchild_id, @rank2:=@rank2+1 as rank FROM grandchild INNER JOIN child ON child.child_id=grandchild.child_id WHERE parent_id=new_parent_id) gc2 ON gc1.rank=gc2.rank 
    INNER JOIN greatgrandchild ggc ON gc1.grandchild_id=ggc.grandchild_id;