純數(shù)據(jù)庫(kù)表實(shí)現(xiàn)方式
首先創(chuàng)建一張表:
CREATE TABLE `methodLock` (
`id` int(11) NOT NULL AUTO_INCREMENT COMMENT '主鍵',
`method_name` varchar(64) NOT NULL DEFAULT '' COMMENT '鎖定的方法名',
`desc` varchar(1024) NOT NULL DEFAULT '備注信息',
`update_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '保存數(shù)據(jù)時(shí)間祭玉,自動(dòng)生成',
PRIMARY KEY (`id`),
UNIQUE KEY `uidx_method_name` (`method_name `) USING BTREE
) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='鎖定中的方法';
當(dāng)我們想要鎖住某個(gè)方法時(shí)抒线,執(zhí)行以下SQL:
insert into methodLock(method_name,desc) values (‘method_name’,‘desc’)
因?yàn)槲覀儗?duì)method_name做了唯一性約束,這里如果有多個(gè)請(qǐng)求同時(shí)提交到數(shù)據(jù)庫(kù)的話,數(shù)據(jù)庫(kù)會(huì)保證只有一個(gè)操作可以成功其爵,那么我們就可以認(rèn)為操作成功的那個(gè)線程獲得了該方法的鎖,可以執(zhí)行方法體內(nèi)容。
當(dāng)方法執(zhí)行完畢之后侨歉,想要釋放鎖的話,需要執(zhí)行以下Sql:
delete from methodLock where method_name ='method_name'
上面這種簡(jiǎn)單的實(shí)現(xiàn)有以下幾個(gè)問(wèn)題:
- 這把鎖強(qiáng)依賴(lài)數(shù)據(jù)庫(kù)的可用性揩魂,數(shù)據(jù)庫(kù)是一個(gè)單點(diǎn)幽邓,一旦數(shù)據(jù)庫(kù)掛掉,會(huì)導(dǎo)致業(yè)務(wù)系統(tǒng)不可用火脉。
- 這把鎖沒(méi)有失效時(shí)間颊艳,一旦解鎖操作失敗,就會(huì)導(dǎo)致鎖記錄一直在數(shù)據(jù)庫(kù)中忘分,其他線程無(wú)法再獲得到鎖棋枕。
- 這把鎖只能是非阻塞的,因?yàn)閿?shù)據(jù)的insert操作妒峦,一旦插入失敗就會(huì)直接報(bào)錯(cuò)重斑。沒(méi)有獲得鎖的線程并不會(huì)進(jìn)入排隊(duì)隊(duì)列,要想再次獲得鎖就要再次觸發(fā)獲得鎖操作肯骇。
- 這把鎖是不可重入的窥浪,同一個(gè)線程在沒(méi)有釋放鎖之前無(wú)法再次獲得該鎖。因?yàn)閿?shù)據(jù)中數(shù)據(jù)已經(jīng)存在了笛丙。
當(dāng)然漾脂,我們也可以有其他方式解決上面的問(wèn)題。
- 數(shù)據(jù)庫(kù)是單點(diǎn)胚鸯?搞兩個(gè)數(shù)據(jù)庫(kù)骨稿,數(shù)據(jù)之前雙向同步。一旦掛掉快速切換到備庫(kù)上。
- 沒(méi)有失效時(shí)間坦冠?只要做一個(gè)定時(shí)任務(wù)形耗,每隔一定時(shí)間把數(shù)據(jù)庫(kù)中的超時(shí)數(shù)據(jù)清理一遍
- 非阻塞的?搞一個(gè)while循環(huán)辙浑,直到insert成功再返回成功激涤。
- 非重入的?在數(shù)據(jù)庫(kù)表中加個(gè)字段判呕,記錄當(dāng)前獲得鎖的機(jī)器的主機(jī)信息和線程信息倦踢,那么下次再獲取鎖的時(shí)候先查詢(xún)數(shù)據(jù)庫(kù),如果當(dāng)前機(jī)器的主機(jī)信息和線程信息在數(shù)據(jù)庫(kù)可以查到的話侠草,直接把鎖分配給他就可以了硼一。
基于數(shù)據(jù)庫(kù)排他鎖
除了可以通過(guò)增刪操作數(shù)據(jù)表中的記錄以外,其實(shí)還可以借助數(shù)據(jù)中自帶的鎖來(lái)實(shí)現(xiàn)分布式的鎖梦抢。
我們還用剛剛創(chuàng)建的那張數(shù)據(jù)庫(kù)表般贼。可以通過(guò)數(shù)據(jù)庫(kù)的排他鎖來(lái)實(shí)現(xiàn)分布式鎖奥吩。
基于MySql的InnoDB引擎哼蛆,可以使用以下方法來(lái)實(shí)現(xiàn)加鎖操作:
public boolean lock(){
connection.setAutoCommit(false)
while(true){
try{
result = select * from methodLock where method_name=xxx for update;
if(result==null){
return true;
}
}catch(Exception e){
}
sleep(1000);
}
return false;
}
在查詢(xún)語(yǔ)句后面增加for update,數(shù)據(jù)庫(kù)會(huì)在查詢(xún)過(guò)程中給數(shù)據(jù)庫(kù)表增加排他鎖霞赫。當(dāng)某條記錄被加上排他鎖之后腮介,其他線程無(wú)法再在該行記錄上增加排他鎖。(這里再多提一句端衰,InnoDB引擎在加鎖的時(shí)候叠洗,只有通過(guò)索引進(jìn)行檢索的時(shí)候才會(huì)使用行級(jí)鎖,否則會(huì)使用表級(jí)鎖旅东。這里我們希望使用行級(jí)鎖灭抑,就要給method_name添加索引,值得注意的是抵代,這個(gè)索引一定要?jiǎng)?chuàng)建成唯一索引腾节,否則會(huì)出現(xiàn)多個(gè)重載方法之間無(wú)法同時(shí)被訪問(wèn)的問(wèn)題。重載方法的話建議把參數(shù)類(lèi)型也加上)
我們可以認(rèn)為獲得排它鎖的線程即可獲得分布式鎖荤牍,當(dāng)獲取到鎖之后案腺,可以執(zhí)行方法的業(yè)務(wù)邏輯,執(zhí)行完方法之后康吵,再通過(guò)以下方法解鎖:
public void unlock(){
connection.commit();
}
這里還可能存在另外一個(gè)問(wèn)題劈榨,雖然我們對(duì)method_name 使用了唯一索引,并且顯示使用for update來(lái)使用行級(jí)鎖晦嵌。但是同辣,MySql會(huì)對(duì)查詢(xún)進(jìn)行優(yōu)化拷姿,即便在條件中使用了索引字段,但是否使用索引來(lái)檢索數(shù)據(jù)是由 MySQL 通過(guò)判斷不同執(zhí)行計(jì)劃的代價(jià)來(lái)決定的邑闺,如果 MySQL 認(rèn)為全表掃效率更高,比如對(duì)一些很小的表棕兼,它就不會(huì)使用索引陡舅,這種情況下 InnoDB 將使用表鎖,而不是行鎖伴挚。如果發(fā)生這種情況就悲劇了靶衍。。茎芋。
還有一個(gè)問(wèn)題颅眶,就是我們要使用排他鎖來(lái)進(jìn)行分布式鎖的lock,那么一個(gè)排他鎖長(zhǎng)時(shí)間不提交田弥,就會(huì)占用數(shù)據(jù)庫(kù)連接涛酗。一旦類(lèi)似的連接變得多了,就可能把數(shù)據(jù)庫(kù)連接池?fù)伪?/p>