項目中經常會用到自增id,比如uid,下面為大家介紹下利用mysql事務特性實現並發安全的自增ID,感興趣的朋友可以參考下
項目中經常會用到自增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都得弄一張表來維護,這也是缺點。 具體使用中如何處理,就看自己的選擇了。