Mysql并以原子方式执行存储过程或以原子方式选择更新

问题描述

在 MysqL 中，我有两个并发进程需要读取一些行并根据条件更新标志。我已经用事务创建了一个存储过程，但问题是有时这两个过程更新相同的行。

我有一个表状态，我想读取保留标志为真的 15 行，然后更新那些将标志保留为 False 的行。更新后的行必须返回给客户端。

我的存储过程是：

CREATE DEFINER=`user`@`%` PROCEDURE `get_reserved`()
BEGIN
DECLARE tmpProfilePageId bigint;
DECLARE finished INTEGER DEFAULT 0;

DECLARE curProfilePage CURSOR FOR 
    SELECT ProfilePageId 
    FROM Status
    WHERE Reserved is false and ((timestampdiff(HOUR,UpdatedTime,Now()) >= 23) or UpdatedTime is NULL)
    ORDER BY UpdatedTime ASC
    LIMIT 15;
DECLARE CONTINUE HANDLER 
    FOR NOT FOUND SET finished = 1;
    
DECLARE EXIT HANDLER FOR sqlEXCEPTION ROLLBACK;
DECLARE EXIT HANDLER FOR sqlWARNING ROLLBACK;

START TRANSACTION;

DROP TEMPORARY TABLE IF EXISTS TmpAdsProfile;
CREATE TEMPORARY TABLE TmpAdsProfile(Id INT PRIMARY KEY AUTO_INCREMENT,ProfilePageId BIGINT);

OPEN curProfilePage;

getProfilePage: LOOP
    FETCH curProfilePage INTO tmpProfilePageId;
    IF finished = 1 THEN LEAVE getProfilePage;
    END IF;
    UPDATE StatusSET Reserved = true WHERE ProfilePageId = tmpProfilePageId;
    INSERT INTO TmpAdsProfile (ProfilePageId) VALUES (tmpProfilePageId);
END LOOP getProfilePage;

CLOSE curProfilePage;

SELECT ProfilePageId FROM TmpAdsProfile;

COMMIT;

END

无论如何，如果我执行两个调用此存储过程的并发进程，有时它们会更新相同的行。

如何以原子方式执行存储过程？

解决方法

稍微简化一下并使用 FOR UPDATE。这将锁定您要更改的行，直到您提交事务。您可以完全摆脱光标。这样的东西，没有调试！

START TRANSACTION;

CREATE OR REPLACE TEMPORARY TABLE TmpAdsProfile AS
SELECT ProfilePageId 
  FROM Status
 WHERE Reserved IS false 
   AND ((timestampdiff(HOUR,UpdatedTime,NOW()) >= 23) OR UpdatedTime IS NULL)
 ORDER BY UpdatedTime ASC
 LIMIT 15 
   FOR UPDATE; 

 UPDATE Status SET Reserved = true 
  WHERE ProfilePageId IN (SELECT ProfilePageId FROM TmpAdsProfile);
 
COMMIT;

SELECT ProfilePageId FROM TmpAdsProfile;

那个临时表中只有十五行。所以索引和 PK 以及所有不需要的东西。因此，您可以使用 CREATE ... AS SELECT ... 一次性创建和填充表格。

并且，考虑重铸您的 UpdatedTime 过滤器，以便它可以使用索引。

AND (UpdatedTime <= NOW() - INTERVAL 23 HOUR OR UpdatedTime IS NULL)

SELECT 查询的合适索引是

CREATE INDEX status_update ON Status (Reserved,ProfilePageId);

您的 SELECT 操作越快，您的事务花费的时间就越少，因此您的整体性能就会越好。

database-concurrency mysql sql sql transactions