在 MySQL 中做一个可靠的计数器

Question

我想在使用 MySQL 数据库的 PHP 编写的应用程序中创建一个计数器。用户给出一个字符串作为输入。此字符串表示数据库记录的键。它存储一个整数计数器值。用户取回增加的值。

这就是我目前的做法：

// connect to database
try {
    $db = new PDO('mysql:host='.$dbhost.';dbname='.$dbname, $dbuser, $dbpasswd);
    $db->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
    $db->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_ASSOC);
} catch(Exception $ex) {
    echo "DB Error: ".$ex->getMessage();
    exit;
}

// update and get new value
try {
    $db->beginTransaction();
    $nc = updateNC($db, $grundsig, $library);
    $db->commit();
} catch(Exception $ex) {
    $db->rollBack();
    $err = 'DB Error: '.$ex->getMessage();
}

// [...]

function updateNC($db, $grundsig, $library) {

    // increment counter ("nc")
    $query = $db->prepare('
        update numeruscurrens n
        join libraries l using(libid)
        set n.nc = n.nc+1
        where n.grundsig = :grundsig and l.libname = :libname
    ');
    $query->execute([
        ':grundsig' => $grundsig,
        ':libname' => $library
    ]);

    // get new counter value
    $query = $db->prepare('
        select *
        from numeruscurrens n
        join libraries l using(libid)
        where n.grundsig = :grundsig and l.libname = :libname
    ');
    $query->execute([
        ':grundsig' => $grundsig,
        ':libname' => $library
    ]);
    $result = $query->fetchAll();
    return $result[0]['nc'];
}

重要的部分发生在updateNC 函数中。我做了一个更新来增加计数器，我想要新的计数器值。

现在问题出在这两个命令之间，另一个用户可以进行更新。两个用户可能会得到错误的计数器值。

如何可靠地增加一个计数器并为该用户获取新值？

我还考虑过使用像 Redis 这样具有 INCR 命令的键值数据库。但我认为这有点夸张。

Answer 1

如果您想要确保他们将获得立即增加的值，那么您应该使用一种方法在提交事务之前锁定您要更新的行。这可以使用FOR UPDATE 提示和之前的 SELECT 查询来锁定您将要更新的行。

我将只添加 SQL 代码以关注实际问题。

SET @counter = 0;

BEGIN;

SELECT n.nc INTO @counter
FROM numeruscurrens n
JOIN libraries l using(libid)
WHERE n.grundsig = @grundsig and l.libname = @libname
FOR UPDATE;

update numeruscurrens n
join libraries l using(libid)
set n.nc = n.nc+1
where n.grundsig = @grundsig and l.libname = @libname

/* You are always incrementing, aren't you? */
SET @counter = @counter + 1;

SELECT @counter;

COMMIT;

另外请注意，这不是免费的。实际上必须要求将此增量值正确返回给用户，因为此锁定会因行锁定而对性能产生负面影响并可能产生死锁。最后但同样重要的是，确保表使用InnoDB 引擎。 MyIsam 不支持行级锁。