快捷導(dǎo)航

mysql 隊(duì)列實(shí)現(xiàn)并發(fā)讀

更新時(shí)間：2012年04月24日 20:39:01 作者：

隊(duì)列是常用的數(shù)據(jù)結(jié)構(gòu)，基本特點(diǎn)就是先入先出，在事務(wù)處理等方面都要用到它，有的時(shí)候是帶有優(yōu)先級(jí)的隊(duì)列。當(dāng)隊(duì)列存在并發(fā)訪問(wèn)的時(shí)候，比如多線程情況下，就需要鎖機(jī)制來(lái)保證隊(duì)列中的同一個(gè)元素不被多次獲取

一個(gè) MySQL 表可以看作是一個(gè)隊(duì)列，每一行為一個(gè)元素。每次查詢得到滿足某個(gè)條件的最前面的一行，并將它從表中刪除或者改變它的狀態(tài)，使得下次查詢不會(huì)得到它。在沒(méi)有并發(fā)訪問(wèn)的情況下，簡(jiǎn)單地用 SELECT 得到一行，再用UPDATE(或者DELETE)語(yǔ)句修改之，就可以實(shí)現(xiàn)。

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

 
SELECT * FROM targets WHERE status='C' LIMIT 1; 
UPDATE targets SET status='D' WHERE id='id'; 

如果有并發(fā)訪問(wèn)，在SELECT和UPDATE語(yǔ)句之間可能會(huì)存在其他地SELECT查詢，導(dǎo)致同一行被取出多次。為了保證在并發(fā)情況下仍然能正常工作，一種思路是使用數(shù)據(jù)庫(kù)地鎖來(lái)防止，就像在多線程環(huán)境下所做地一樣。總之，要是的查詢和修改為一個(gè)原子操作，不被其它的訪問(wèn)干擾。MySQL 5 支持存儲(chǔ)過(guò)程，可以用它來(lái)實(shí)現(xiàn)。
單條 UPDATE 語(yǔ)句應(yīng)該原子操作的，可以利用這個(gè)特性來(lái)保證并發(fā)訪問(wèn)情況下隊(duì)列的正常工作。每次取元素時(shí)，先用 UPDATE 修改符合條件的第一行，然后再得到該行?？上?UPDATE 語(yǔ)句沒(méi)有返回值，重新用普通的SELECT的話又很難找到剛被改過(guò)的那條記錄。
這里用到一個(gè)小技巧：在 UPDATE 時(shí)加上 id=LAST_INSERT_ID(id)，再用 SELECT LAST_INSERT_ID() 即可得到剛修改的那條記錄的id。還有一個(gè)問(wèn)題，當(dāng)表中不存在符合條件的記錄，導(dǎo)致 UPDATE 失敗時(shí)，LAST_INSERT_ID() 會(huì)保留原來(lái)地值不變，因而不能區(qū)分隊(duì)列中是否還有元素。
ROW_COUNT() 返回上一個(gè)語(yǔ)句影響的行數(shù)，把它作為 SELECT 的一個(gè)條件，可以幫助解決這個(gè)問(wèn)題。
最后，支持并發(fā)訪問(wèn)的完整解決方案為：

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

 
UPDATE targets SET status='D', id=LAST_INSERT_ID(id) WHERE status='C' LIMIT 1; 
SELECT * FROM targets WHERE ROW_COUNT()>0 and id=LAST_INSERT_ID(); 

更新：在實(shí)現(xiàn)帶優(yōu)先級(jí)的隊(duì)列時(shí)這種方法有問(wèn)題，帶有 ORDER BY ... 條件的 UPDATE 語(yǔ)句非常慢，例如：

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

UPDATE targets SET status='D' WHERE status='C' ORDER BY schedule ASC LIMIT 1;

而單獨(dú)查詢和更新則是很快的：

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

 
SELECT id FROM targets WHERE status='C' ORDER BY schedule ASC LIMIT 1; 
UPDATE targets SET status='D' WHERE id='id'; 

原來(lái)這是MySQL的Bug-12915，一年多以前提出來(lái)的，雖然關(guān)閉了，卻只解決了部分問(wèn)題，尚不支持WHERE，見(jiàn)MySQL 5.0.15 的 Changlog。無(wú)奈，上面這種巧妙的方法也沒(méi)有實(shí)用價(jià)值了。
最后采用了一種折衷方案，如下：

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

 
UPDATE targets, (SELECT id FROM targets WHERE status='C' AND schedule<CURRENT_TIMESTAMP ORDER BY schedule ASC LIMIT 1) tmp SET status='D' WHERE targets.id=LAST_INSERT_ID(tmp.id); 
SELECT * FROM targets WHERE ROW_COUNT()>0 and id=LAST_INSERT_ID(); 

您可能感興趣的文章: