使用java和redis不加锁的速率限制器

Question

我想了解限速器是怎么工作的。

我在博客中看到的是：

public void makeApiCall(String ip){
    Long currentTime=Timestamp timestamp = System.currentTimeMillis();

    String key=ip+":"+currentTime;

    Integer count=redisClient.get(key);

    if(count!=null && count > 10){
         throw LimitExceededException();
    }
    else{
         redisClient.incr(key,1);
         callApi();
    }
}

到现在为止，我忽略了删除旧的钥匙。我不明白这个索伦是怎么工作的？正如上面所说，在多线程环境中，代码将在一秒钟内完成10个以上的api调用。它只能通过将redisclient.get（key）同步到callApi（）代码来解决。

我从

假设在当前的第二个中已经服务了9个请求，现在同时出现了5个新的请求，所有这些新的线程在这5个线程中的任何一个执行“else”块之前调用redisclient.get（key）。因此，对于每个线程，计数将是9，else块将被执行，并且incr将被调用5次，对于每个线程，将调用api。

Answer 1

实际上，代码确实容易受到竞争条件（和内存膨胀，因为您没有考虑过期）的影响。解决这一问题的方法基本上有两种：使用watch的multi/exec事务，或者使用Lua脚本的eval。

假设您使用Jedis作为Java客户机，那么类似于下面这样的操作就可以处理事务：

public void makeApiCall(String ip){
    Long currentTime=Timestamp timestamp = System.currentTimeMillis();

    String key=ip+":"+currentTime;

    redisClient.watch(key);
    Integer count=redisClient.get(key);

    if(count!=null && count > 10){
         throw LimitExceededException();
    }
    else{
         Transaction t = redisClient.multi();
         t.incr(key,1);
         List<Object> resp = t.exec();
         if(resp != null){
             callApi();
         }
    }
}

Lua则是另一回事，但基本上假设您提供以下脚本（整个密钥名(ip+ts)，只要您的代码在获得OK后继续使用callapi，它就会执行同样的操作：

local count = redis.call('GET', KEYS[1])
if count and tonumber(count) > 10 then
    return redis.error('count exceeded')
else
    redis.call('INCR', KEYS[1])
    redis.call('EXPIRE', KEYS[1], 10)
    return 'OK'
end

注意，使用Lua您不需要观察更改，因为整个脚本是原子的。

最后，您在文档中提到的计数器模式似乎缺少watch这一点--我已经提交了一个PR来纠正这一点(https://github.com/antirez/redis-doc/pull/888)。