网络安全

位置:IT落伍者 >> 网络安全 >> 浏览文章

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


发布日期:2024年05月23日
 
利用mysql事务特性实现并发安全的自增ID示例
项目中经常会用到自增id比如uid最简单的方法就是用直接用数据库提供的AUTO_INCREMENT但是如果用户量非常大几千万几亿然后需要分表存储的时候呢这种方案就搞不定了所以最好有一个全局的自增ID的生成器不管是否分表都能从生成器中获取到全局自增的ID

实现方法应该有很多不过所有的方案都需要解决一个问题就是保证在高并发的情景下数据获取依然正确每次获取的ID都不会重复

这里我分享两种利用mysql的innodb的事务特性来实现的方案一种是实现过了的另一种没有试验过不过应该也能走的通

先介绍第一种在数据库中单独设置一张表来存储ID表有两个字段一个是种类吧一个就是ID

复制代码 代码如下:

CREATE TABLE auto_id(

idname varchar() NOT NULL DEFAULT

id bigint() NOT NULL DEFAULT COMMENT

primary key(idname)

)ENGINE=Innodb DEFAULT CHARSET=utf;

接下来是一个存储过程

复制代码 代码如下:

delimiter //

drop procedure if exists get_increment_id;

create procedure get_increment_id(in idname_in varchar() 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(idnameid) value(idname_in small_in);

set id_out=small_in;

else

update maibo_auto_id set id=id+ where idname=idname_in;

set id_out=oldid+;

end if;

commit;

end;

//

重点是这句select id into oldid from maibo_auto_id where idname=idname_in for update会给相关数据加一个独占锁定这时候别的进程如果来读取该条记录就会进入等待等待这个进程commit之后再继续这样就保证了在并发的情况下不同的进程不会取到相同的值

如果你的前端是用php实现的

只需执行如下两个sql就可以获取到这个small参数是定义的是从多少开始自增

复制代码 代码如下:

$sql = call get_increment_id({$key} {$small} @id);

$ret = $db>getData(select @id);

还有另外一种方法就是利用mysql的auto_increment

先创建一张表表里边只有一个自增字段

复制代码 代码如下:

create table test(

`id` int() NOT NULL AUTO_INCREMENT COMMENT id

primary key (id)

)ENGINE=MyISAM AUTO_INCREMENT= DEFAULT CHARSET=utf;

通过如下两条sql

复制代码 代码如下:

UPDATE test SET id = LAST_INSERT_ID(id + );

SELECT LAST_INSERT_ID();

也能解决问题 LAST_INSERT_ID是不用查表的而且只针对当前连接也就是说别的连接的更新不会影响到当前连接的取值

这样可能每个ID都得弄一张表来维护这也是缺点

具体使用中如何处理就看自己的选择了

上一篇:类型安全的EventHandlerList

下一篇:彻底清除无法删除的文件 360安全卫士“文件粉碎机”一键搞定