解鎖redis鎖的正確姿勢(shì)

redis是php的好朋友，在php寫業(yè)務(wù)過(guò)程中，有時(shí)候會(huì)使用到鎖的概念，同時(shí)只能有一個(gè)人可以操作某個(gè)行為。這個(gè)時(shí)候我們就要用到鎖。鎖的方式有好幾種，php不能在內(nèi)存中用鎖，不能使用zookeeper加鎖，使用數(shù)據(jù)庫(kù)做鎖又消耗比較大，這個(gè)時(shí)候我們一般會(huì)選用redis做鎖機(jī)制。

setnx

鎖在redis中最簡(jiǎn)單的數(shù)據(jù)結(jié)構(gòu)就是string。最早的時(shí)候，上鎖的操作一般使用setnx，這個(gè)命令是當(dāng):lock不存在的時(shí)候set一個(gè)val，或許你還會(huì)記得使用expire來(lái)增加鎖的過(guò)期，解鎖操作就是使用del命令，偽代碼如下：

if (Redis::setnx("my:lock", 1)) {
  Redis::expire("my:lock", 10);
  // ... do something

  Redis::del("my:lock")
}

這里其實(shí)是有問(wèn)題的，問(wèn)題就在于setnx和expire中間如果遇到crash等行為，可能這個(gè)lock就不會(huì)被釋放了。于是進(jìn)一步的優(yōu)化方案可能是在lock中存儲(chǔ)timestamp。判斷timestamp的長(zhǎng)短。

set

現(xiàn)在官方建議直接使用set來(lái)實(shí)現(xiàn)鎖。我們可以使用set命令來(lái)替代setnx，就是下面這個(gè)樣子

if (Redis::set("my:lock", 1, "nx", "ex", 10)) {
  ... do something

  Redis::del("my:lock")
}

上面的代碼把my:lock設(shè)置為1，當(dāng)且僅當(dāng)這個(gè)lock不存在的時(shí)候，設(shè)置完成之后設(shè)置過(guò)期時(shí)間為10。

獲取鎖的機(jī)制是對(duì)了，但是刪除鎖的機(jī)制直接使用del是不對(duì)的。因?yàn)橛锌赡軐?dǎo)致誤刪別人的鎖的情況。

比如，這個(gè)鎖我上了10s，但是我處理的時(shí)間比10s更長(zhǎng)，到了10s，這個(gè)鎖自動(dòng)過(guò)期了，被別人取走了，并且對(duì)它重新上鎖了。那么這個(gè)時(shí)候，我再調(diào)用Redis::del就是刪除別人建立的鎖了。

官方對(duì)解鎖的命令也有建議，建議使用lua腳本，先進(jìn)行g(shù)et，再進(jìn)行del

程序變成：

$token = rand(1, 100000);

function lock() {
  return Redis::set("my:lock", $token, "nx", "ex", 10);
}

function unlock() {
  $script = `
if redis.call("get",KEYS[1]) == ARGV[1]
then
  return redis.call("del",KEYS[1])
else
  return 0
end  
  `
  return Redis::eval($script, "my:lock", $token)
}

if (lock()) {
  // do something

  unlock();
}

這里的token是一個(gè)隨機(jī)數(shù)，當(dāng)lock的時(shí)候，往redis的my:lock中存的是這個(gè)token，unlock的時(shí)候，先get一下lock中的token，如果和我要?jiǎng)h除的token是一致的，說(shuō)明這個(gè)鎖是之前我set的，否則的話，說(shuō)明這個(gè)鎖已經(jīng)過(guò)期，是別人set的，我就不應(yīng)該對(duì)它進(jìn)行任何操作。

所以：不要再使用setnx，直接使用set進(jìn)行鎖實(shí)現(xiàn)。

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

最新評(píng)論