基于SELECT COUNT结果分配角色并插入用户，是否适合使用事务？

为什么需要事务？ #

你的操作包含两个强关联的步骤：

1. 查询用户表的记录数，以此决定新用户的角色（第一条用户设为admin，后续为普通user）
2. 插入新用户数据

如果不使用事务，当多个请求同时执行时，会出现竞态条件：比如两个请求同时查到用户数为0，都会把自己的用户角色设为1（admin），最后你的系统会出现两个管理员，这显然不符合设计预期。

事务能保证这两个操作的原子性——要么查询+插入都成功，要么都失败；同时配合数据库的隔离级别（比如默认的REPEATABLE READ），能避免并发查询时的不一致问题。

优化后的代码实现 #

try { 
    // 连接数据库
    require_once 'connect.php';
    
    // 开启事务
    $conn->beginTransaction();

    // 查询用户表记录数
    $querySelect = $conn->prepare("SELECT COUNT(userId) FROM users");
    $querySelect->execute();
    // 直接获取count值，无需转JSON
    $userCount = $querySelect->fetchColumn();
    // 简化角色判断逻辑
    $sUserRole = $userCount === 0 ? 1 : 2;

    // 插入新用户
    $queryInsert = $conn->prepare("INSERT INTO users ( userName, firstName, lastName, password, image, userRoles_roleId ) VALUES ( :userName, :firstName, :lastName, :password, :image, :userRole )");
    $queryInsert->bindParam( ':userName' , $sUserName );
    $queryInsert->bindParam( ':firstName' , $sFirstName );
    $queryInsert->bindParam( ':lastName' , $sLastName );
    $queryInsert->bindParam( ':password' , $sPassword );
    $queryInsert->bindParam( ':image' , $sImage );
    $queryInsert->bindParam( ':userRole' , $sUserRole );
    
    $bResult = $queryInsert->execute();

    if ($bResult) {
        // 事务提交
        $conn->commit();
        echo '{"status":"ok"}';
    } else {
        // 插入失败，回滚事务
        $conn->rollBack();
        echo '{"status":"error","message":"Failed to insert user"}';
    }

} catch (Exception $e) {
    // 捕获异常时回滚事务
    if ($conn->inTransaction()) {
        $conn->rollBack();
    }
    echo '{"status":"error","message":"' . $e->getMessage() . '"}';
}

关键改进点 #

• 事务控制：通过beginTransaction()、commit()、rollBack()确保操作的原子性，避免并发问题
• 简化结果处理：用fetchColumn()直接获取count值，比转JSON再判断更高效可靠
• 完善错误处理：插入失败或捕获异常时都会回滚事务，同时返回更具体的错误信息，方便调试
• 注意数据库引擎：确保你的users表使用的是支持事务的引擎（比如MySQL的InnoDB，MyISAM不支持事务）

内容的提问来源于stack exchange，提问作者codeDragon