欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

基于Redis的限流器的实现(示例讲解)

程序员文章站 2022-04-10 10:14:16
1 概述 系统中的接口通常都有限流,比如 70次/秒 ,如何保证我们的接口的调用次数在超过第三方接口限流的时候快速失败呢?这时候就需要限流器了。下面是笔者用redis实现...

1 概述

系统中的接口通常都有限流,比如 70次/秒 ,如何保证我们的接口的调用次数在超过第三方接口限流的时候快速失败呢?这时候就需要限流器了。下面是笔者用redis实现限流器的流程图。

基于Redis的限流器的实现(示例讲解)

2 代码

/**
 * 获取限流权限
 * @param key
 * @param millisecond 毫秒数
 * @param limitcount 限流次数
 * @return
 */
public static boolean getcurrentlimiting(string key, long millisecond, integer limitcount){
  try {
    boolean currentlimitinglock = redisdistributelockutil.lock(key);
    if(currentlimitinglock){
      long llen = redisclient.llen(getcurrentlimitingrediskey(key));
      if(llen < limitcount){
        redisclient.lpush(getcurrentlimitingrediskey(key), dateutil.current(false) + "");
        return true;
      }else{
        long lasttime = long.parselong(redisclient.lindex(getcurrentlimitingrediskey(key), -1));
        if((dateutil.current(false) - lasttime) >= millisecond){
          redisclient.lpush(getcurrentlimitingrediskey(key), dateutil.current(false) + "");
          redisclient.ltrim(getcurrentlimitingrediskey(key), 0, limitcount);
          return true;
        }
      }
    }
  }finally {
    redisdistributelockutil.unlock(key);
  }
  return false;
}

以上这篇基于redis的限流器的实现(示例讲解)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。