如何在 MySQL 上正确循环存储函数？

Question

我在正确执行一个非常简单的存储过程时遇到了一些困难。 考虑以下文章表片段：

id    replaced_by     baseID
 1              2          0
 2              3          0
 3              0          0

一个简单的分层表，使用写时复制。编辑文章时，当前文章的 Replaced_by 字段将设置为其新副本的 ID。

我添加了一个 baseID 字段，将来应该存储文章的 baseID。 在我上面的示例中，有一篇文章（例如 id 3）。它的baseID将是1。

为了获取baseID，我创建了以下存储过程：

DELIMITER $$

CREATE FUNCTION getBaseID(articleID INT) RETURNS INT
BEGIN
    DECLARE x INT;
    DECLARE y INT;
    SET x = articleID;
    sloop:LOOP
        SELECT id INTO y FROM article WHERE replaced_by_articleID = x;
        IF y IS NOT NULL THEN
            SET x = y;
            ITERATE sloop;
        ELSE
            LEAVE sloop;
        END IF;  
    END LOOP;
    RETURN x;
END $$

DELIMITER ;

看起来很简单，直到我实际使用以下方式调用该函数：

SELECT getBaseID(3);

我希望该函数返回1。我什至愿意理解它可以花一秒钟的时间。 相反，机器的 CPU 上升到 100% (mysqld)。

我什至使用 REPEAT .. UNTIL 和 WHILE .. DO 重写了相同的函数，并具有相同的最终结果。

谁能解释一下为什么我的CPU在进入循环时会上升100%？

旁注：我只是想赢得时间。我在 PHP 中创建了完全相同的函数，它执行得不错，但我们猜测 MySQL 可以做得更快一些。我们需要筛选大约 1800 万条记录。我能节省的任何一点时间都是值得的。

预先感谢您的任何帮助和/或指示。

---

已解决的SQL：

DELIMITER $$

CREATE FUNCTION getBaseID(articleID INT) RETURNS INT
BEGIN
    DECLARE x INT;
    DECLARE y INT;
    SET x = articleID;
    sloop:LOOP
        SET y = NULL;
        SELECT id INTO y FROM article WHERE replaced_by_articleID = x;
        IF y IS NULL THEN
            LEAVE sloop;
        END IF;  
        SET x = y;
        ITERATE sloop;
    END LOOP;
    RETURN x;
END $$

DELIMITER ;

Answer 1

来自 mysql ：

如果查询没有返回行，则会出现错误代码 1329 的警告（无数据），并且变量值保持不变

因此，当未找到给定 x 的记录时，就会出现无限循环（y 保持不变）
尝试使用 SET y = (SELECT id ....) 或在 select 语句之前添加 SET y = null （它应该是循环中的第一个语句）

Answer 2

感觉您可能缺少 Replaced_by 列上的索引。如果 Replaced_by 上没有索引，则每次迭代都会进行全表扫描。

ALTER TABLE article ADD INDEX (replaced_by);

之前确保该行存在

DELIMITER $

CREATE FUNCTION getBaseID(articleID INT) RETURNS INT
BEGIN
    DECLARE x INT;
    DECLARE y INT;
    SET x = articleID;
    sloop:LOOP
        SELECT COUNT(1) INTO y FROM article WHERE replaced_by = x;
        IF y > 0 THEN
            SELECT id INTO y FROM article WHERE replaced_by = x;
            SET x = y;
        ELSE
            LEAVE sloop;
        END IF;  
    END LOOP;
    RETURN x;
END $

DELIMITER ;

您还应该在检索两倍的 SQL 调用

，但安全总比后悔好。尝试一下！