利用mysql事务特性实现并发安全的自增ID示例
时间：2020-12-17 01:40:56|栏目：Mysql|点击： 次
项目中经常会用到自增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都得弄一张表来维护，这也是缺点。
具体使用中如何处理，就看自己的选择了。
实现方法应该有很多，不过所有的方案都需要解决一个问题，就是保证在高并发的情景下，数据获取依然正确，每次获取的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都得弄一张表来维护，这也是缺点。
具体使用中如何处理，就看自己的选择了。
