最新redis实现分布式限流 结合Lua脚本，Java开发笔试题编程题

在Redis中实现分布式限流通常使用Lua脚本来确保操作的原子性。以下是一个简单的Java代码示例，展示了如何使用Jedis客户端执行Lua脚本以实现限流。

import redis.clients.jedis.Jedis;
 
public class RedisRateLimiter {
    private static final String LUA_SCRIPT = 
            "local key = KEYS[1] " +
            "local limit = tonumber(ARGV[1]) " +
            "local current = tonumber(redis.call('get', key) or '0') " +
            "if current + 1 > limit then return 0 else " +
            "redis.call('INCRBY', key, '1') " +
            "redis.call('EXPIRE', key, '10') " +
            "return 1 end";
 
    private Jedis jedis;
 
    public RedisRateLimiter(Jedis jedis) {
        this.jedis = jedis;
    }
 
    public boolean isAllowed(String userId, int limit) {
        Long isAllowed = (Long) jedis.eval(LUA_SCRIPT, 1, "user:" + userId, String.valueOf(limit));
        return isAllowed == 1L;
    }
 
    public static void main(String[] args) {
        Jedis jedis = new Jedis("localhost", 6379);
        RedisRateLimiter rateLimiter = new RedisRateLimiter(jedis);
 
        // 假设用户ID是"user1"，限制10秒内只能访问5次
        boolean allowed = rateLimiter.isAllowed("user1", 5);
        System.out.println("Is allowed: " + allowed);
 
        jedis.close();
    }
}

在这个例子中，isAllowed 方法使用了传入的用户ID和限制次数来调用Lua脚本。Lua脚本会检查当前计数器的值，如果加1后超过限制，则不允许访问，并返回0。如果未超过限制，则允许访问一次，并重置过期时间为10秒。

注意：在实际应用中，你可能需要更复杂的限流策略，例如滑动窗口或漏桶算法，但这个简单的例子展示了如何使用Lua脚本和Redis来实现基本的限流功能。