利用mysql事務(wù)特性實(shí)現(xiàn)并發(fā)安全的自增ID示例

更新時(shí)間：2013年11月27日 17:44:44 作者：

項(xiàng)目中經(jīng)常會(huì)用到自增id，比如uid，下面為大家介紹下利用mysql事務(wù)特性實(shí)現(xiàn)并發(fā)安全的自增ID，感興趣的朋友可以參考下

項(xiàng)目中經(jīng)常會(huì)用到自增id，比如uid，最簡(jiǎn)單的方法就是用直接用數(shù)據(jù)庫(kù)提供的AUTO_INCREMENT，但是如果用戶量非常大，幾千萬(wàn)，幾億然后需要分表存儲(chǔ)的時(shí)候呢，這種方案就搞不定了，所以最好有一個(gè)全局的自增ID的生成器，不管是否分表，都能從生成器中獲取到全局自增的ID。

實(shí)現(xiàn)方法應(yīng)該有很多，不過(guò)所有的方案都需要解決一個(gè)問(wèn)題，就是保證在高并發(fā)的情景下，數(shù)據(jù)獲取依然正確，每次獲取的ID都不會(huì)重復(fù)。

這里我分享兩種利用mysql的innodb的事務(wù)特性來(lái)實(shí)現(xiàn)的方案，一種是實(shí)現(xiàn)過(guò)了的，另一種沒(méi)有試驗(yàn)過(guò)，不過(guò)應(yīng)該也能走的通。

先介紹第一種，在數(shù)據(jù)庫(kù)中單獨(dú)設(shè)置一張表，來(lái)存儲(chǔ)ID，表有兩個(gè)字段，一個(gè)是種類吧，一個(gè)就是ID：

復(fù)制代碼代碼如下:

 
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; 

接下來(lái)是一個(gè)存儲(chǔ)過(guò)程：

復(fù)制代碼代碼如下:

 
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; 
// 

重點(diǎn)是這句，select id into oldid from maibo_auto_id where idname=idname_in for update，會(huì)給相關(guān)數(shù)據(jù)加一個(gè)獨(dú)占鎖定，這時(shí)候別的進(jìn)程如果來(lái)讀取該條記錄，就會(huì)進(jìn)入等待，等待這個(gè)進(jìn)程commit之后，再繼續(xù)，這樣就保證了在并發(fā)的情況下，不同的進(jìn)程不會(huì)取到相同的值。

如果你的前端是用php實(shí)現(xiàn)的。

只需執(zhí)行如下兩個(gè)sql，就可以獲取到，這個(gè)small參數(shù)是定義的是從多少開(kāi)始自增

復(fù)制代碼代碼如下:

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

還有另外一種方法，就是利用mysql的auto_increment。

先創(chuàng)建一張表，表里邊只有一個(gè)自增字段：

復(fù)制代碼代碼如下:

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

通過(guò)如下兩條sql：

復(fù)制代碼代碼如下:

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

也能解決問(wèn)題， LAST_INSERT_ID是不用查表的，而且只針對(duì)當(dāng)前連接，也就是說(shuō)別的連接的更新不會(huì)影響到當(dāng)前連接的取值。

這樣可能每個(gè)ID都得弄一張表來(lái)維護(hù)，這也是缺點(diǎn)。

具體使用中如何處理，就看自己的選擇了。

您可能感興趣的文章: