JAVA调用

  1. private boolean checkAllowed(String routeId) {
  2.     RateLimiterConfig rateLimiterConfig = configMap.get(routeId);
  3.     int qps = rateLimiterConfig.getRate();
  4.     long now = System.currentTimeMillis();
  5.     String key = String.format(RATE_LIMITER_ITEM_KEY_PATTERN, routeId);
  7.     Object result = jedisTemplate.evalsha(rateLimiterScriptId, 1, key, String.valueOf(now), String.valueOf(qps));
  8.     return Integer.valueOf(result.toString()) > 0;
  9. }

限流lua脚本

  1. local key = KEYS[1]
  2. local curTime = tonumber(ARGV[1])
  3. local qps = tonumber(ARGV[2])
  4. local millis = math.modf(1000 / qps)
  6. local array = redis.call('hmget', key, 'curToken', 'lastTime')
  7. local curToken = array[1]
  8. local lastTime = array[2]
  9. if (type(curToken) ~= 'string') then
  10.     curToken = 1
  11.     lastTime = curTime
  12.     redis.call('hmset', key, 'curToken', curToken, 'lastTime', lastTime)
  13. else
  14.     curToken = tonumber(curToken)
  15.     lastTime = tonumber(lastTime)
  16. end
  18. local tokenToAdd = math.modf((curTime - lastTime) / millis)
  19. local totalToken = curToken + tokenToAdd
  20. local overload = false
  22. if (totalToken > qps) then
  23.     totalToken = qps
  24.     overload = true
  25. end
  27. if (totalToken > 0) then
  28.     redis.call('hset', key, 'curToken', totalToken - 1)
  30.     if (tokenToAdd > 0) then
  31.         if (overload) then
  32.             lastTime = curTime
  33.         else
  34.             lastTime = lastTime + tokenToAdd * millis
  35.         end
  36.         redis.call('hset', key, 'lastTime', lastTime)
  37.     end
  39.     return 1
  40. else
  41.     return 0
  42. end