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

2022-11-12 09:36:15
内容摘要
这篇文章主要为大家详细介绍了利用MySQL事务特性实现并发安全的自增ID示例,具有一定的参考价值,可以用来参考一下。 对此感兴趣的朋友,看看idc笔记做的技术笔记!项目中经常会用
文章正文

这篇文章主要为大家详细介绍了利用MySQL事务特性实现并发安全的自增ID示例,具有一定的参考价值,可以用来参考一下。

对此感兴趣的朋友,看看idc笔记做的技术笔记!

项目中经常会用到自增id,比如uid,最简单的方法就是用直接用数据库提供的AUTO_INCREMENT,但是如果用户量非常大,几千万,几亿然后需要分表存储的时候呢,这种方案就搞不定了,所以最好有一个全局的自增ID的生成器,不管是否分表,都能从生成器中获取到全局自增的ID。实现方法应该有很多,不过所有的方案都需要解决一个问题,就是保证在高并发的情景下,数据获取依然正确,每次获取的ID都不会重复。这里我分享两种利用mysql的innodb的事务特性来实现的方案,一种是实现过了的,另一种没有试验过,不过应该也能走的通。先介绍第一种,在数据库中单独设置一张表,来存储ID,表有两个字段,一个是种类吧,一个就是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; 
接下来是一个存储过程:

代码如下:

 
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; 
// 
重点是这句,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(11) NOT NULL AUTO_INCREMENT COMMENT 'id', 
primary key (id) 
)ENGINE=MyISAM AUTO_INCREMENT=1 DEFAULT CHARSET=utf8; 
通过如下两条sql:

代码如下:

 
UPDATE test SET id = LAST_INSERT_ID(id + 1); 
SELECT LAST_INSERT_ID(); 
也能解决问题, LAST_INSERT_ID是不用查表的,而且只针对当前连接,也就是说别的连接的更新不会影响到当前连接的取值。这样可能每个ID都得弄一张表来维护,这也是缺点。具体使用中如何处理,就看自己的选择了。

注:关于利用MySQL事务特性实现并发安全的自增ID示例的内容就先介绍到这里,更多相关文章的可以留意

代码注释

作者:喵哥笔记

IDC笔记

学的不仅是技术,更是梦想!