JAVA基于Redis实现计数器限流的使用示例
更新时间：2023年09月25日 11:01:32 作者：拾荒的小海螺
计数器法是限流算法里最简单也是最容易实现的一种算法,本文主要介绍了JAVA基于Redis实现计数器限流的使用示例,具有一定的参考价值,感兴趣的可以了解一下
1、简述
在现实世界中可能会出现服务器被虚假请求轰炸的情况，因此您可能希望控制这种虚假的请求。
一些实际使用情形可能如下所示：
API配额管理-作为提供者，您可能希望根据用户的付款情况限制向服务器发出API请求的速率。这可以在客户端或服务端实现。
安全性-防止DDOS攻击。
成本控制–这对服务方甚至客户方来说都不是必需的。如果某个组件以非常高的速率发出一个事件，它可能有助于控制它，它可能有助于控制从客户端发送的请求。
常见的限流算法：令牌桶算法， 漏桶算法。比较成熟的有分布式hystrix, sentinel，还有guava高并发限流ratelimiter。本文主要是介绍Redis如何对指定的key进行计数限流的。
2、引用和配置
引用redis的maven包，包括客户端连接插件jedis。
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-data-redis</artifactId> </dependency> <dependency> <groupId>redis.clients</groupId> <artifactId>jedis</artifactId> </dependency>
在application.yml配置中添加Redis服务端配置：
spring:
#redis配置#
redis:
database: 0
host: 192.168.254.128
port: 6379
password: 123456
timeout: 5000
jedis:
pool:
max-active: 8
max-wait: 5000
max-idle: 8
min-idle: 13、Config配置
添加redis configuration配置bean和redis存储json转换。
@Configuration
public class MyRedisConfig {
@Value("${spring.redis.host}")
private String host;
@Value("${spring.redis.password}")
private String password;
@Value("${spring.redis.port}")
private int port;
@Value("${spring.redis.database}")
private int database;
//@SuppressWarnings("all")
@Bean
public StringRedisTemplate redisTemplate(RedisConnectionFactory factory){
StringRedisTemplate template = new StringRedisTemplate(factory);
Jackson2JsonRedisSerializer jackson2JsonRedisSerializer = new Jackson2JsonRedisSerializer(Object.class);
ObjectMapper om = new ObjectMapper();
om.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY);
om.enableDefaultTyping(ObjectMapper.DefaultTyping.NON_FINAL);
jackson2JsonRedisSerializer.setObjectMapper(om);
RedisSerializer stringSerializer = new StringRedisSerializer();
template.setKeySerializer(stringSerializer);
template.setValueSerializer(jackson2JsonRedisSerializer);
template.setHashKeySerializer(stringSerializer);
template.setHashValueSerializer(jackson2JsonRedisSerializer);
template.afterPropertiesSet();
return template;
}
@Bean
public JedisConnectionFactory jedisConnectionFactory() {
RedisStandaloneConfiguration configuration = new RedisStandaloneConfiguration();
configuration.setHostName(host);
configuration.setPassword(RedisPassword.of(password));
configuration.setPort(port);
configuration.setDatabase(database);
JedisConnectionFactory jedisConnectionFactory = new JedisConnectionFactory(configuration);
jedisConnectionFactory.getPoolConfig().setMaxIdle(30);
jedisConnectionFactory.getPoolConfig().setMinIdle(10);
return jedisConnectionFactory;
}
}4、添加组件
添加自定义的限速拦截器组件，到时候拦截器会根据该组件注释来拦截对应的接口请求，实现跟业务解耦。
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface RequestLimit {
/**
* 调用方唯一key的名字
*
* @return
*/
String name();
/**
* 限制访问次数
* @return
*/
int limitTimes();
/**
* 限制时长，也就是计数器的过期时间
*
* @return
*/
long timeout();
/**
* 限制时长单位
*
* @return
*/
TimeUnit timeUnit();
}- name ：表示请求方唯一的身份参数，如userId,token等。
- limitTimes ：表示限制访问次数，也就是他在指定时间内可以访问多少次。
- timeout：表示限制访问次数的有效期，一分钟还是一个小时。
- timeUnit：表示限速实际的单位，秒、分钟、小时等
5、拦截器
添加拦截器实现HandlerInterceptor 来实现对添加@RequestLimit 进行拦截处理当前请求在规定的时间是否超出。当前采用的是redis的BoundValueOperations 来计算请求次数在规定的时间内是否异常。
/**
* 请求限流拦截器
*/
@Slf4j
@Component
public class RequestLimitInterceptor implements HandlerInterceptor {
@Autowired
private RedisTemplate redisTemplate;
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
if(handler instanceof HandlerMethod){
HandlerMethod handlerMethod = (HandlerMethod) handler;
//判断接口是否添加requestLimit
if(handlerMethod.hasMethodAnnotation(RequestLimit.class)){
RequestLimit requestLimit = handlerMethod.getMethod().getAnnotation(RequestLimit.class);
JSONObject object = new JSONObject();
String token = request.getParameter(requestLimit.name());
response.setContentType("text/json;charset=utf-8");
object.put("timestamp", System.currentTimeMillis());
BoundValueOperations<String, Integer> boundValueOperations = redisTemplate.boundValueOps(token);
if(StringUtils.isEmpty(token)){
object.put("result", "token is invalid");
response.getWriter().print(JSON.toJSONString(object));
} else if(checkLimit(token,requestLimit)){
object.put("result","token is success,请求成功");
long expire = boundValueOperations.getExpire();
return true;
}else {
object.put("result", "达到访问次数上限，禁止访问!");
response.getWriter().print(JSON.toJSONString(object));
}
return false;
}
}
return true;
}
/**
* 限速校验
* @param token
* @param limit
* @return
*/
private Boolean checkLimit(String token, RequestLimit limit){
BoundValueOperations<String , Integer> boundValueOperations = redisTemplate.boundValueOps(token);
Integer count = boundValueOperations.get();
if(Objects.isNull(count)){
redisTemplate.boundValueOps(token).set(1,limit.timeout(), limit.timeUnit());
}else if(count > limit.limitTimes()){
return Boolean.FALSE;
}else {
redisTemplate.boundValueOps(token).set(++count, boundValueOperations.getExpire(),limit.timeUnit());
}
return Boolean.TRUE;
}
}实现拦截器后我们需要将当前拦截器添加到WebMvcConfigurer拦截器中：
@Configuration
public class MyWebConfig implements WebMvcConfigurer {
@Autowired
private RequestLimitInterceptor requestLimitInterceptor;
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(requestLimitInterceptor).addPathPatterns("/**");
WebMvcConfigurer.super.addInterceptors(registry);
}
}这样我们就完成了简易的拦截限速器，我们在请求的接口前添加@RequestLimit就可以限速了：
/**
* 分页查询
* @param param
* @return
*/
@RequestLimit(name = "token", limitTimes = 20, timeout = 60, timeUnit = TimeUnit.SECONDS)
@GetMapping("/page")
public R page(TagParam param) {
Query query = param.toQuery();
PageInfo<Tag> pageInfo = tagService.page(query);
return R.ok().put("pageInfo", pageInfo);
}6、BoundValueOperations 使用
6.1 BoundValueOperations
就是一个绑定key的对象，我们可以通过这个对象来进行与key相关的操作
BoundValueOperations boundValueOps = redisTemplate.boundValueOps("token");6.2 set(V value)
给绑定键重新设置值(如果没有值，则会添加这个值)
boundValueOps.set("token");6.3 get()
获取绑定键的值。
String str = (String) boundValueOps.get(); System.out.println(str);
6.4 set(V value, long timeout, TimeUnit unit)
给绑定键设置新值并设置过期时间
boundValueOps.set("token",30, TimeUnit.SECONDS);6.5 getAndSet(V value)
如果有这个值则获取没有则设置
String oldValue = (String) boundValueOps.getAndSet("token");
String newValue = (String) boundValueOps.get();6.6 increment(double delta)和increment(long delta)
它是Redis的自增长键，前提是绑定值的类型是double或long类型。increment是单线程的，所以它是安全的。
BoundValueOperations boundValueOps = redisTemplate.boundValueOps("token");
boundValueOps.set(1);
System.out.println(boundValueOps.get());
boundValueOps.increment(1);
System.out.println(boundValueOps.get());注意！使用该方法，需要使用StringRedisSerializer序列化器才能使用increment方法，否则会报错
7、代码地址
不管代码实现方式如何，还是要自己动手来实现才能体验设计的思想，让自己成长的更快，理解的更透彻。
代码地址：https://gitee.com/lhdxhl/redis.git
参考文章：https://zhuanlan.zhihu.com/p/427906048
到此这篇关于JAVA基于Redis实现计数器限流的使用示例的文章就介绍到这了,更多相关JAVA Redis计数器限流内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了使用spring boot 整合kafka,延迟启动消费者的操作，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-08-08
这篇文章主要介绍了Java基础之练习打印三角形,文中有非常详细的代码示例,对正在学习java基础的小伙伴们有非常好的帮助,需要的朋友可以参考下2021-04-04
MyBatis-Plus updateById更新不了空字符串或null的解决方法
本文主要介绍了MyBatis-Plus updateById更新不了空字符串或null的解决方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-03-03
这篇文章主要介绍了SpringBoot 如何从容器中获取对象，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-08-08
这篇文章主要介绍了java内部类原理与用法,结合实例形式分析了Java内部类的概念、原理、分类及相关使用技巧,需要的朋友可以参考下2019-05-05
这篇文章主要介绍了深入理解java泛型Generic,文中有非常详细的代码示例,对正在学习java的小伙伴们有非常好的帮助,需要的朋友可以参考下2021-05-05
IOC也是Spring的核心之一了，之前学的时候是采用xml配置文件的方式去实现的，后来其中也多少穿插了几个注解，但是没有说完全采用注解实现。那么这篇文章就和大家分享一下，全部采用注解来实现IOC + DI2022-10-10
这篇文章主要介绍了java四舍五入使java保留2位小数示例，大家参考使用2013-12-12
这篇文章主要给大家介绍了关于win10下定时运行与开机自启动jar包的相关资料，文中通过图文介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-11-11
这篇文章主要介绍了Java并发编程：volatile关键字详细解析，对学习volatile关键字有一定的认识，有需要的可以了解一下。2016-11-11
最新评论