redis 實現限速器的幾種方式。 GET + INCR + EXPIRE 先獲取 key 的當前值,如果沒有超出限制再執行 INCR 增1,如果 key 不存在,使用 redis 的事務初始化 key 和過期時間。 偽代碼: count = redis.GET(key) if redis retu ...
redis 實現限速器的幾種方式。
GET + INCR + EXPIRE
先獲取 key 的當前值,如果沒有超出限制再執行 INCR 增1,如果 key 不存在,使用 redis 的事務初始化 key 和過期時間。
偽代碼:
count = redis.GET(key) if redis return nil { redis.MULTI redis.INCR(key) redis.EXPIRE(key, expire_time) redis.EXEC count = 1 } if count > limit { return 超出限制 } else { redis.INCR(key) }
高併發下的問題:
如果同時10個併發程式執行 GET 返回了 nil, 那麼這10個併發程式都會執行 redis 的事務將 key 增一,但每個程式的 count 值都為1,如果 limit 設置的值小於10,那麼真正執行的程式就超過限制了。如果執行完事務後再查一次 redis 賦值給 count,那麼每個程式可能都會返回10,從而沒有程式能夠繼續執行。
key 已經存在的情況下,先 GET 後 INCR 的邏輯也可能會出現實際執行的程式數多於 limit 的情況。
INCR + EXPIRE
先 INCR, 如果值為1說明是 key 剛設置的,此時再執行 EXPIRE
偽代碼:
count = redis.INCR(key) if count == 1 { redis.EXPIRE(key, expire_time) } if count > limit { return 超出限制 }
慎用
如果 INCR 之後程式掛掉了,沒有執行 EXPIRE, 那麼這個 key 就沒有過期時間了,具體的影響視需求而定。
lua腳本
local current current = redis.call("incr",KEYS[1]) if tonumber(current) == 1 then redis.call("expire",KEYS[1],1) end
更多學習內容請訪問:
八重櫻:騰訊T3-T4標準精品PHP架構師教程目錄大全,只要你看完保證薪資上升一個臺階(持續更新)