基于springboot 整合Redis实现令牌桶分布式API限流

最新推荐文章于 2023-03-06 14:40:14 发布

fenyudelushang

最新推荐文章于 2023-03-06 14:40:14 发布

阅读量1.5k

点赞数

分类专栏： redis springboot java8 文章标签： redis spring boot 分布式 java 缓存

本文链接：https://blog.youkuaiyun.com/fenyudelushang/article/details/120894815

版权

这里简单说一下用Redis实现令牌桶算法的方法。

用户每一次访问请求的时候，从redis中获取一个令牌，如果拿到令牌了，就说明没有超出限制，就通过这个请求，相反如果拿不到令牌就拦截这个请求或者返回错误信息，下面是代码实现部分。

1.定义Redis限流规则类

public final class RRateLimiter {
	private String rediskey = "limit";
	private Long all_conut = 20L;//令牌的总数量
	private Long conut = 10L;//生产的令牌数量
	private Long time = 5L;//产生的令牌的时间间隔
	private TimeUnit timetype = TimeUnit.SECONDS;
	private ListOperations<String,Object> redis;
	private void start() {
		new Thread(() -> {
			try {
				while (true) {
					timetype.sleep(time);
					if(redis.size(rediskey)<all_conut)
					LongStream.range(0,conut).forEach(x->redis.rightPush(rediskey,UUID.randomUUID().toString().replace("-","")));
				}
			} catch(Exception e){}
		}).start();
	}
	public RRateLimiter(ListOperations<String, Object> list, String rediskey, Long conut, Long time,
			TimeUnit timetype) {
		th