項目中經常會用到自增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都得弄一張表來維護,這也是缺點。
具體使用中如何處理,就看自己的選擇了。