1
原本以為自己對redis命令還蠻熟悉的,各種資料模型各種基于redis的騷操作,但是最近在使用redis的scan的命令式卻踩了一個坑,頓時發覺自己原來對redis的游標理解的很有限,所以記錄下這個踩坑的程序,背景如下:
公司因為redis服務器記憶體吃緊,需要洗掉一些無用的沒有設定過期時間的key,大概有500多w的key,雖然key的數目聽起來挺嚇人,但是自己玩redis也有年頭了,這種事還不是手到擒來?
當時想了下,具體方案是通過lua腳本來過濾出500w的key,然后進行洗掉動作,lua腳本在redis server上執行,執行速度快,執行一批只需要和redis server建立一次連接,篩選出來key,然后一次刪1w,然后通過shell腳本回圈個500次就能刪完所有的,以前通過lua腳本做過類似批量更新的操作,3w一次也是秒級的,基本不會造成redis的阻塞,這樣算起來,10分鐘就能搞定500w的key,
然后,我就開始直接寫lua腳本,首先是篩選,
用過redis的人,肯定知道redis是單執行緒作業的,肯定不能用keys命令來篩選,因為keys命令會一次性進行全盤搜索,會造成redis的阻塞,從而會影響正常業務的命令執行,
500w資料量的key,只能增量迭代來進行,redis提供了scan命令,就是用于增量迭代的,這個命令可以每次回傳少量的元素,所以這個命令十分適合用來處理大的資料集的迭代,可以用于生產環境,

scan命令會回傳一個陣列,第一項為游標的位置,第二項是key的串列,如果游標到達了末尾,第一項會回傳0,
2
所以我寫的第一版的lua腳本如下:
local c = 0
local resp = redis.call('SCAN',c,'MATCH','authToken*','COUNT',10000)
c = tonumber(resp[1])
local dataList = resp[2]
for i=1,#dataList do
local d = dataList[i]
local ttl = redis.call('TTL',d)
if ttl == -1 then
redis.call('DEL',d)
end
end
if c==0 then
return 'all finished'
else
return 'end'
end
在本地的測驗redis環境中,通過執行以下命令mock了20w的測驗資料:
eval "for i = 1, 200000 do redis.call('SET','authToken_' .. i,i) end" 0
然后執行script load命令上傳lua腳本得到SHA值,然后執行evalsha去執行得到的SHA值來運行,具體程序如下:

我每刪1w資料,執行下dbsize(因為這是我本地的redis,里面只有mock的資料,dbsize也就等同于這個前綴key的數量了),
奇怪的是,前面幾行都是正常的,但是到了第三次的時候,dbsize變成了16999,多刪了1個,我也沒太在意,但是最后在dbsize還剩下124204個的時候,數量就不動了,之后無論再執行多少遍,數量還依舊是124204個,
隨即我直接運行scan命令:

發現游標雖然沒有到達末尾,但是key的串列卻是空的,
這個結果讓我懵逼了一段時間,我仔細檢查了lua腳本,沒有問題啊,難道是redis的scan命令有bug?難道我理解的有問題?
我再去翻看redis的命令檔案對count選項的解釋:

經過詳細研讀,發現count選項所指定的回傳數量還不是一定的,雖然知道可能是count的問題,但無奈檔案的解釋實在難以很通俗的理解,依舊不知道具體問題在哪
3
后來經過某個小伙伴的提示,看到了另外一篇對于scan命令count選項通俗的解釋:

看完之后恍然大悟,原來count選項后面跟的數字并不是意味著每次回傳的元素數量,而是scan命令每次遍歷字典槽的數量
我scan執行的時候每一次都是從游標0的位置開始遍歷,而并不是每一個字典槽里都存放著我所需要篩選的資料,這就造成了我最后的一個現象:雖然我count后面跟的是10000,但是實際redis從開頭往下遍歷了10000個字典槽后,發現沒有資料槽存放著我所需要的資料,所以我最后的dbsize數量永遠停留在了124204個,
所以在使用scan命令的時候,如果需要迭代的遍歷,需要每次呼叫都需要使用上一次這個呼叫回傳的游標作為該次呼叫的游標引數,以此來延續之前的迭代程序,
至此,心中的疑惑就此解開,改了一版lua:
local c = tonumber(ARGV[1])
local resp = redis.call('SCAN',c,'MATCH','authToken*','COUNT',10000)
c = tonumber(resp[1])
local dataList = resp[2]
for i=1,#dataList do
local d = dataList[i]
local ttl = redis.call('TTL',d)
if ttl == -1 then
redis.call('DEL',d)
end
end
return c
在本地上傳后執行:


可以看到,scan命令沒法完全保證每次篩選的數量完全等同于給定的count,但是整個迭代卻很好的延續下去了,最后也得到了游標回傳0,也就是到了末尾,至此,測驗資料20w被全部刪完,
這段lua只要在套上shell進行回圈就可以直接在生產上跑了,經過估算大概在12分鐘左右能洗掉掉500w的資料,
知其然,知其所以然,雖然scan命令以前也曾玩過,但是的確不知道其中的細節,況且檔案的翻譯也不是那么的準確,以至于自己在面對錯誤的結果時整整浪費了近1個多小時的時間,記錄下來,加深理解,
4.聯系作者
微信關注 「jishuyuanren」或者掃描以下二維碼獲取更多干貨

轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/140455.html
標籤:Java
