CodeIgniter中并发注册的邮箱去重策略：利用表锁解决竞态条件

<?php
defined('BASEPATH') OR exit('No direct script access allowed');

class UserModel extends CI_Model {

    public function __construct() {
        parent::__construct();
        $this->load->database();
    }

    /**
     * 尝试注册新用户，通过表级写锁防止并发重复邮箱。
     *
     * @param array $userData 包含用户信息的数组，至少应包含 'email' 键。
     * @return bool 注册成功返回 true，否则返回 false。
     */
    public function registerUserWithLock($userData) {
        // 确保邮箱数据存在
        if (!isset($userData['email'])) {
            log_message('error', '注册用户时缺少邮箱信息。');
            return false;
        }

        // 启动数据库事务，虽然表锁独立于事务，但结合使用可增强操作的原子性
        $this->db->trans_start();

        try {
            // 1. 获取 'users' 表的写锁
            // 注意：LOCK TABLES 是 MySQL 特有的语法。对于其他数据库，需要使用其相应的锁机制。
            // 例如，PostgreSQL 可能使用 SELECT ... FOR UPDATE。
            $this->db->query("LOCK TABLES users WRITE");

            // 2. 在锁的保护下，检查邮箱是否已存在
            $this->db->where('email', $userData['email']);
            $query = $this->db->get('users');

            if ($query->num_rows() > 0) {
                // 邮箱已存在，释放锁并回滚事务
                $this->db->query("UNLOCK TABLES");
                $this->db->trans_rollback();
                log_message('info', '尝试注册的邮箱 ' . $userData['email'] . ' 已存在。');
                return false; // 邮箱已注册
            }

            // 3. 邮箱不存在，插入新用户数据
            $insert_result = $this->db->insert('users', $userData);

            // 4. 释放写锁
            $this->db->query("UNLOCK TABLES");

            // 完成事务
            $this->db->trans_complete();

            if ($this->db->trans_status() === FALSE) {
                // 事务失败，可能由于其他数据库错误
                log_message('error', '注册用户事务失败。');
                return false;
            }

            return $insert_result; // 返回插入结果
        } catch (Exception $e) {
            // 捕获异常，确保在异常发生时也能释放锁并回滚事务
            $this->db->query("UNLOCK TABLES");
            $this->db->trans_rollback();
            log_message('error', '注册用户时发生异常: ' . $e->getMessage());
            return false;
        }
    }
}

<?php
defined('BASEPATH') OR exit('No direct script access allowed');

class Auth extends CI_Controller {

    public function __construct() {
        parent::__construct();
        $this->load->model('UserModel');
        $this->load->helper('form');
        $this->load->library('form_validation');
    }

    public function register() {
        // 设置表单验证规则
        $this->form_validation->set_rules('email', 'Email', 'required|valid_email');
        $this->form_validation->set_rules('password', 'Password', 'required|min_length[6]');
        // ... 其他字段验证

        if ($this->form_validation->run() == FALSE) {
            // 验证失败，显示注册表单
            $this->load->view('register_form');
        } else {
            $userData = array(
                'email' => $this->input->post('email'),
                'password' => password_hash($this->input->post('password'), PASSWORD_DEFAULT), // 密码哈希
                // ... 其他用户数据
            );

            if ($this->UserModel->registerUserWithLock($userData)) {
                // 注册成功
                echo "注册成功！";
            } else {
                // 注册失败，可能是邮箱已存在或数据库错误
                echo "注册失败，请检查邮箱是否已被注册或稍后再试。";
            }
        }
    }
}