使用redis中的zset实现滑动窗口限流

使用redis和zset实现滑动窗口限流

文章目录

Zset

Redis的ZSet(有序集合)可以很好地用来实现滑动窗口限流。滑动窗口限流是一种常见的流量控制方法,它限制了在一定时间窗口内的请求数量。下面是使用Redis ZSet实现滑动窗口限流的一个简单示例:

初始化一个ZSet:其中包含所有用户的ID和时间戳。

java 复制代码
ZSet<String> zset = redisTemplate.opsForZSet().create("rateLimiter");

添加元素到ZSet:当用户发起请求时,将当前时间戳和用户ID作为元素添加到ZSet中。

java 复制代码
long currentTimeMillis = System.currentTimeMillis();
String userId = "user1";
redisTemplate.opsForZSet().add("rateLimiter", userId, currentTimeMillis);

删除过期的元素:为了保持滑动窗口的大小,需要删除超出时间窗口范围的元素。例如,如果滑动窗口的大小为60秒,那么需要删除60秒之前添加的元素。

java 复制代码
long windowSizeInSeconds = 60;
long currentTimeMillis = System.currentTimeMillis();
// 获取ZSet中所有元素
List<ZSetElement<String>> elements = redisTemplate.opsForZSet().reverseRangeWithScores("rateLimiter", 0, -1);
for (ZSetElement<String> element : elements) {
    long elementTimestamp = element.getScore();
    if (currentTimeMillis - elementTimestamp > windowSizeInSeconds * 1000) {
        redisTemplate.opsForZSet().remove("rateLimiter", element.getValue());
    }
}

检查是否超过限制:在添加新元素后,检查ZSet的大小是否超过限制。如果超过限制,则拒绝请求。

java 复制代码
int limit = 100; // 每分钟的请求限制
long size = redisTemplate.opsForZSet().size("rateLimiter");
if (size >= limit) {
    // 超过限制,拒绝请求...
}

注意,以上代码是基于Java的Spring Data Redis实现,如果你使用其他语言的Redis客户端,代码可能会有所不同,但基本的思路是相同的。此外,这个简单的实现没有考虑分布式环境下的限流,这需要额外的同步机制。

拓展补充

java 复制代码
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

@Component
public class ApiCallCounter {

    private static final String API_CALLS = "api_calls:";

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    public void incrementApiCallCount(String apiName) {
        String key = API_CALLS + apiName + ":current";
        redisTemplate.opsForValue().increment(key);
    }
}

在上述代码中,我们定义了一个ApiCallCounter类,用于计数接口调用量。当接口被调用时,我们使用incrementApiCallCount方法增加计数。该方法使用RedisTemplateopsForValue().increment方法对指定键进行递增操作。我们使用一个包含API名称和时间戳的键来存储每分钟的调用量。例如,如果API名称为exampleApi并且当前时间是2023年7月19日10点05分,则键将是api_calls:exampleApi:current:202307191005

相关推荐
傻啦嘿哟28 分钟前
Python正则表达式:用“模式密码“解锁复杂字符串
linux·数据库·mysql
绿算技术2 小时前
存储新势力:助力DeepSeek一体机
人工智能·科技·缓存·fpga开发
辰哥单片机设计2 小时前
JW01三合一传感器详解(STM32)
数据库·mongodb
小刘同学++2 小时前
Qt使用 SQLite 数据库的基本方法
数据库·qt·sqlite
施嘉伟5 小时前
Oracle 11g RAC ASM磁盘组剔盘、加盘实施过程
数据库·oracle
尤物程序猿5 小时前
【2025面试Java常问八股之redis】zset数据结构的实现,跳表和B+树的对比
数据结构·redis·面试
橘猫云计算机设计6 小时前
springboot基于hadoop的酷狗音乐爬虫大数据分析可视化系统(源码+lw+部署文档+讲解),源码可白嫖!
数据库·hadoop·spring boot·爬虫·python·数据分析·毕业设计
卓怡学长7 小时前
w304基于HTML5的民谣网站的设计与实现
java·前端·数据库·spring boot·spring·html5
冰^7 小时前
MySQL VS SQL Server:优缺点全解析
数据库·数据仓库·redis·sql·mysql·json·数据库开发
电商数据girl7 小时前
产品经理对于电商接口的梳理||电商接口文档梳理与接入
大数据·数据库·python·自动化·产品经理