redis是php的好朋友,在php寫業務過程中,有時候會使用到鎖的概念,同時只能有一個人可以操作某個行為。這個時候我們就要用到鎖。鎖的方式有好幾種,php不能在記憶體中用鎖,不能使用zookeeper加鎖,使用資料庫做鎖又消耗比較大,這個時候我們一般會選用redis做鎖機制。
鎖在redis中最簡單的資料結構就是string。最早的時候,上鎖的操作一般使用setnx,這個命令是當: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就不會被釋放了。於是進一步的優化方案可能是在lock中儲存timestamp。判斷timestamp的長短。
現在官方建議直接使用set來實現鎖。我們可以使用set命令來替代setnx,就是下面這個樣子
if (Redis::set("my:lock", 1, "nx", "ex", 10)) {
... do something
Redis::del("my:lock")
}
上面的程式碼把my:lock設定為1,當且僅當這個lock不存在的時候,設定完成之後設定過期時間為10。
獲取鎖的機制是對了,但是刪除鎖的機制直接使用del是不對的。因為有可能導致誤刪別人的鎖的情況。
比如,這個鎖我上了10s,但是我處理的時間比10s更長,到了10s,這個鎖自動過期了,被別人取走了,並且對它重新上鎖了。那麼這個時候,我再呼叫Redis::del就是刪除別人建立的鎖了。
官方對解鎖的命令也有建議,建議使用lua指令碼,先進行get,再進行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是一個隨機數,當lock的時候,往redis的my:lock中存的是這個token,unlock的時候,先get一下lock中的token,如果和我要刪除的token是一致的,說明這個鎖是之前我set的,否則的話,說明這個鎖已經過期,是別人set的,我就不應該對它進行任何操作。
所以:不要再使用setnx,直接使用set進行鎖實現。
本作品採用《CC 協議》,轉載必須註明作者和本文連結