如何验证随机值是否存在于数据库并插入唯一随机值

你的代码核心思路是对的，但有几个关键问题导致它无法正确实现「生成唯一随机值并插入数据库」的需求，我来逐一拆解并给出修复方案：

原代码的主要问题 #

• 已废弃的数据库扩展：你使用的mysql_*函数在PHP 7及以上版本已经被彻底移除，不仅存在安全隐患，还会直接导致代码运行报错，必须改用mysqli或PDO这类现代数据库扩展。
• 缺少循环验证逻辑：当检测到重复值时，你只调用了random()生成新值，但没有重新执行查询验证和插入操作，相当于只生成了新值却没做后续处理，无法保证最终插入的是唯一值。
• SQL注入风险：直接把变量拼接到SQL语句中，没有做任何转义处理，存在被SQL注入攻击的风险。
• 变量处理不严谨：函数内部的变量作用域虽然不影响返回值，但整体代码的变量复用逻辑不够清晰。

修复后的代码示例（使用mysqli） #

我用mysqli扩展重写了代码，加入循环验证逻辑，同时解决了SQL注入和唯一性保证的问题：

// 数据库连接（替换为你的实际连接信息）
$conn = mysqli_connect('localhost', '你的用户名', '你的密码', '你的数据库名');
if (!$conn) {
    die("数据库连接失败: " . mysqli_connect_error());
}

// 生成6位随机推荐码的函数（PHP7+推荐用random_int更安全）
function generateReferCode() {
    return random_int(100000, 999999);
}

// 循环直到插入唯一值
do {
    $referCode = generateReferCode();
    // 使用预处理语句防止SQL注入
    $checkStmt = mysqli_prepare($conn, "SELECT us_refer_code FROM sign_up WHERE us_refer_code = ?");
    mysqli_stmt_bind_param($checkStmt, "i", $referCode);
    mysqli_stmt_execute($checkStmt);
    mysqli_stmt_store_result($checkStmt);
    $isDuplicate = mysqli_stmt_num_rows($checkStmt) > 0;
    mysqli_stmt_close($checkStmt);

    // 如果不存在则执行插入
    if (!$isDuplicate) {
        $insertStmt = mysqli_prepare($conn, "INSERT INTO sign_up(us_refer_code) VALUES(?)");
        mysqli_stmt_bind_param($insertStmt, "i", $referCode);
        $insertSuccess = mysqli_stmt_execute($insertStmt);
        mysqli_stmt_close($insertStmt);
        
        if ($insertSuccess) {
            echo "插入成功，推荐码：$referCode";
            break; // 插入成功后退出循环
        } else {
            echo "插入失败: " . mysqli_error($conn);
            break;
        }
    }
} while ($isDuplicate); // 存在重复则继续生成新值

// 关闭数据库连接
mysqli_close($conn);

额外优化建议 #

1. 给字段添加唯一索引：在数据库的sign_up表中，给us_refer_code字段添加唯一约束，这样即使出现极端的竞态条件（比如两个请求同时生成了相同的码），数据库也会直接拒绝重复插入，从底层保证唯一性：

   ALTER TABLE sign_up ADD UNIQUE INDEX idx_us_refer_code (us_refer_code);
   

2. 限制循环次数：为了避免极端情况下的无限循环（比如几乎所有6位数字都被占用），可以给循环加一个次数限制，比如最多尝试100次，超过则抛出错误提示。
3. 异常处理：如果项目使用PHP的异常机制，可以把数据库操作包裹在try-catch块中，更优雅地处理错误。

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