解鎖redis鎖的正確姿勢

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

setnx

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

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

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

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

set

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

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

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

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

獲取鎖的機制是對了，但是刪除鎖的機制直接使用del是不對的。因為有可能導(dǎo)致誤刪別人的鎖的情況。

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

官方對解鎖的命令也有建議，建議使用lua腳本，先進行g(shù)et，再進行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是一個隨機數(shù)，當(dāng)lock的時候，往redis的my:lock中存的是這個token，unlock的時候，先get一下lock中的token，如果和我要刪除的token是一致的，說明這個鎖是之前我set的，否則的話，說明這個鎖已經(jīng)過期，是別人set的，我就不應(yīng)該對它進行任何操作。

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

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