利用MySQL事务特性实现并发安全的自增ID

南乔
南乔
发布时间:2026-10-14| 4人浏览过
原创

在用户量达到千万或亿级且需分库分表的业务场景中,数据库原生的自增主键(AUTO_INCREMENT)因局限于单表范围,无法再满足全局唯一性的需求。为解决这一痛点,本文详述如何利用 MySQL 的事务机制与特定函数特性,构建高并发环境下的全局自增 ID 生成器,确保 ID 不重复且获取高效。

核心挑战:并发下的ID唯一性保障

设计全局 ID 生成器的核心难点在于“并发安全”。在高并发读取请求下,必须确保不同进程或线程不会同时获取到相同的 ID 值。若缺乏有效的锁机制或原子操作,极易产生 ID 冲突。以下两种方案均基于此目标设计,分别针对不同引擎的特性进行优化。

方案一:基于 InnoDB 事务排他锁的存储过程法

该方法利用 InnoDB 引擎支持行级锁的特性,通过在查询时附加 FOR UPDATE 子句,对正在读取的行施加独占锁(写锁)。当多个事务同时尝试更新同一行记录时,后续事务将进入等待状态,直到前一个事务提交(COMMIT)并释放锁。这种机制天然实现了串行化访问,从而保证了 ID 递增的唯一性与安全性。

第一步:创建 ID 序列维护表

首先建立一张独立的数据表用于存储各类 ID 的当前状态。该表结构简单,仅需两个字段:标识名称(idname)和当前数值(id)。

CREATE TABLE auto_id (
    idname varchar(20) NOT NULL DEFAULT '', 
    id bigint(20) NOT NULL DEFAULT 0 COMMENT '', 
    primary key(idname) 
) ENGINE=Innodb DEFAULT CHARSET=utf8;

第二步:定义带锁定的存储过程

创建存储过程以封装获取 ID 的逻辑。关键步骤在于使用 START TRANSACTION 开启事务,并通过 SELECT ... FOR UPDATE 锁定待操作的记录。随后根据记录是否存在进行初始赋值或递增操作,最后提交事务释放锁。

delimiter //
drop procedure if exists get_increment_id;
create procedure get_increment_id(
    in idname_in varchar(20), 
    in small_in bigint, 
    out id_out bigint
)
begin 
    declare oldid bigint;
    start transaction;
    -- 加独占锁,确保并发安全
    select id into oldid from maibo_auto_id where idname=idname_in for update;
    if oldid is NULL then 
        -- 若不存在则初始化
        insert into maibo_auto_id(idname,id) value(idname_in, small_in); 
        set id_out=small_in;
    else 
        -- 若存在则自增
        update maibo_auto_id set id=id+1 where idname=idname_in; 
        set id_out=oldid+1;
    end if;
    commit;
end;
//

第三步:应用层调用示例

在 PHP 等后端语言中,可通过调用存储过程并查询用户变量来获取生成的 ID。参数 {$key} 为 ID 类别名,{$small} 为起始值。

$sql = "call get_increment_id('{$key}', {$small}, @id)";
$ret = $db->getData("select @id");

方案二:基于 LAST_INSERT_ID 的原子操作法

若更倾向于轻量级的实现方式,可利用 MySQL 内置函数 LAST_INSERT_ID() 的线程安全性与原子性。该函数返回当前连接最新执行 AUTO_INCREMENT 语句产生的 ID 值,且不受其他连接影响。即使不使用显式的 LOCK,也能通过巧妙的 SQL 构造实现并发安全的递增。

第一步:创建 MyISAM 自增表

由于主要依赖自动增量机制,此处可使用 MyISAM 引擎建立仅含自增字段的测试表。

CREATE TABLE test (
    `id` int(11) NOT NULL AUTO_INCREMENT COMMENT 'id', 
    primary key (id) 
) ENGINE=MyISAM AUTO_INCREMENT=1 DEFAULT CHARSET=utf8;

第二步:执行原子递增查询

通过 UPDATE 语句触发 LAST_INSERT_ID() 的行为,使其返回新计算的值,随后立即 SELECT 该返回值。这两步在并发环境下能保证读取到的始终是递增后的正确数值。

UPDATE test SET id = LAST_INSERT_ID(id + 1);
SELECT LAST_INSERT_ID();

方案对比与维护建议

  • InnoDB 方案:优势在于逻辑清晰,可集中管理多种类型的 ID 序列(只需插入不同的 idname),适合需要统一管理的复杂业务。缺点是涉及存储过程调用和锁等待,存在一定的性能开销。
  • LAST_INSERT_ID 方案:优势是速度快,无需显式事务锁,利用了数据库底层原子性。主要缺点是每个 ID 类型可能需要单独的表来维护,或者在多类型混用时需要更复杂的处理,导致表结构膨胀。

具体应用中可根据对并发性能的要求以及代码维护成本来选择最适合的方案。

以上就是利用mysql事务特性实现并发安全的自增ID的详细内容,更多关于全局ID生成策略的资料请关注本站其它相关文章!

相关标签:

免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。

Python 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。