RedisTemplate实现令牌桶限流

本文主要是介绍RedisTemplate实现令牌桶限流，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

令牌桶

Redisson可以实现很多东西,在Redis的基础上,Redisson做了超多的封装，不仅可以用来实现分布式锁，还可以帮助我们实现令牌桶限流。

Ratelimter主要作用就是可以限制调用接口的次数。主要原理就是调用接口之前，需要拥有指定个令牌，限流器每秒会产生X个令牌放入令牌桶，调用接口需要去令牌桶里面拿令牌。如果令牌被其它请求拿完了，那么自然而然，当前请求就调用不到指定的接口。

RateLimter实现限流

@RestController@RequestMapping("/redisTest")public class RedisTestController {@Autowiredprivate Redisson redisson;@GetMapping("/Token")public String testTokenBucket() {RRateLimiter rateLimiter = redisson.getRateLimiter("myRatelimiter");//最大流速 =每10秒钟产生1个令牌rateLimiter.trySetRate(RateType.OVERALL, 1, 10, RateIntervalUnit.SECONDS);//需要1个令牌if (rateLimiter.tryAcquire(1)) {return "令牌桶里面有可使用的令牌";}return "不好意思，请过十秒钟再来~~~~~~~";}}

RedisTemplate实现方式：

话不多说直接上代码，如下：

package com.shop.cyshop.commons.service;/*** 限流器** @date 2024年04月28日 15:02*/import com.shop.cyshop.commons.locker.LockService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;import java.util.concurrent.TimeUnit;@Service
@Slf4j
public class RequestLimitService {private static final String REQUEST_LIMIT_KEY = "request_limit";private static final int WINDOW_SIZE_SECONDS = 6; //每六秒一次请求限制private static final int MAX_REQUESTS = 1;@Autowiredprivate RedisTemplate<String, String> redisTemplate;@Autowiredprivate LockService lockService;public boolean isAllowed() {return lockService.lock(REQUEST_LIMIT_KEY + "::glock", "sys", () -> {long currentTime = System.currentTimeMillis() / 1000; //10long windowStart = currentTime - WINDOW_SIZE_SECONDS;//4// 删除过期的记录redisTemplate.opsForZSet().removeRangeByScore(REQUEST_LIMIT_KEY, 0, windowStart);// 获取当前窗口内的请求数量Long currentRequests = redisTemplate.opsForZSet().count(REQUEST_LIMIT_KEY, windowStart, currentTime);if (currentRequests != null && currentRequests < MAX_REQUESTS) {// 增加请求记录redisTemplate.opsForZSet().add(REQUEST_LIMIT_KEY, String.valueOf(currentTime), currentTime);// 设置过期时间redisTemplate.expire(REQUEST_LIMIT_KEY, WINDOW_SIZE_SECONDS, TimeUnit.SECONDS);return true;}return false;});}
}

这篇关于RedisTemplate实现令牌桶限流的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---