mysql 隊列實現(xiàn)并發(fā)讀

2020-01-19 00:00:40

字體：大中小

供稿：網(wǎng)友

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

復制代碼代碼如下:

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

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

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

復制代碼代碼如下:

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

而單獨查詢和更新則是很快的：

復制代碼代碼如下:

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

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

復制代碼代碼如下:

 
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(); 

上一篇：mysql unsigned 用法及相減出現(xiàn)補數(shù)溢出解決方法

下一篇：mysql中RAND()隨便查詢記錄效率問題和解決辦法分享