php使用lua+redis實現限流,計數器模式,令牌桶模式

這篇文章主要介紹了php使用lua+redis實現限流,計數器模式,令牌桶模式,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨着小編來一起學習學習吧

lua 優點

減少網絡開銷: 不使用 Lua 的代碼需要向 Redis 發送多次請求, 而腳本只需一次即可, 減少網絡傳輸;

原子操作: Redis 將整個腳本作爲一個原子執行, 無需擔心併發, 也就無需事務;

複用: 腳本會永久保存 Redis 中, 其他客戶端可繼續使用.

計數器模式:

利用lua腳本一次性完成處理達到原子性,通過INCR自增計數,判斷是否達到限定值,達到限定值則返回限流,添加key過期時間應該範圍過度

$lua = '
        local i = redis.call("INCR", KEYS[1]) 
        if i > 10 then
          return "wait"
        else
          if i == 1
          then
            redis.call("expire", KEYS[1], KEYS[2])
          end
          return redis.call("get", KEYS[3])
        end
      ';

laravel 請求代碼:

Redis::eval($lua, 3, sprintf(RedisKey::API_LIMIT, $key, $callService['service']), 60, $cache_key);

令牌桶模式

每次請求在桶內拿取一個令牌,有令牌則通過,否則返回,並且按照算法一定的慢慢把令牌放入桶內

$lua = '
        local data = redis.call("get", KEYS[2])
        if data then
        
          local dataJson = cjson.decode(data)
          local newNum = math.min(KEYS[3], math.floor(((dataJson["limitVal"] - 1) + (KEYS[3]/KEYS[5]) * (KEYS[4] - dataJson["limitTime"]))))
          
          if newNum > 0 then
          
            local paramsJson = cjson.encode({limitVal=newNum,limitTime=KEYS[4]})
            redis.call("set", KEYS[2], paramsJson)
            return redis.call("get", KEYS[1])
          
          end
           return "wait"
        end
        
        local paramsJson = cjson.encode({limitVal=KEYS[3],limitTime=KEYS[4]})
        redis.call("set", KEYS[2], paramsJson)
        return redis.call("get", KEYS[1])
      ';
      
      // 1. lua腳本, 2 KEYS數量, 3 查找數據key, 4 限制key, 5 桶內數量, 6 時間戳, 7 過期時間
      Redis::eval(1,2,3,4,5,6,7參數);

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持神馬文庫。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章