脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 数据库 → Redis → Redis接口防刷

详解Redis如何优雅地实现接口防刷

 更新时间：2024年03月12日 10:13:36   作者：shigen01  
这篇文章主要为大家详细介绍了Redis优雅地实现接口防刷的相关知识,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下

背景

最近在学习redis，想到了之前的写的一个案例demo，实现了接口的流量防刷。主要是为了防止爬虫爬取接口，当然可以适用于那些需要进行流控的系统，shigen画了一张草图展示主要的原理和过程：

首先用户请求系统的接口，后端会有一个流控的拦截器，这里边主要是去获得请求的信息，用ip+URI作为redis的key，查询在redis中记录的次数，如果次数大于规定的单位时间的访问次数，就认为这是非法的请求了，就需要拦截了。那具体的代码该怎么设计呢》？下边是一份详细的代码案例。亮点在于：shigen使用了一个注解，支持接口的个性化设置。

代码实现

自定义ReteLimit注解

package main.java.com.shigen.redis.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy,
import java.lang.annotation.Target;

/**
 * @author shigenfu
 * @date 2023/9/6 10:05 上午
 */
 
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface RateLimit {
    
    /**
     * 限定时间
     * 
     * @return 限定时间
     */
     int seconds() default 1;
     /**
     * 最大请求数
     * 
     * @return 最大请求数!
     */
     int max() default 1;
}

其实就是两个核心参数seconds max 规定了单位时间内的最大访问次数。

拦截器rateLimitIntercept

/**
* @author shigenfu
* @date 2023/9/6  10:09  上午
*/

@Component
public class RateLimitIntercept extends HandlerInterceptorAdapter {
    
    @Resource
    private RedisTemplate<String, String> redisTemplate;
    public static final String RATE_LIMIT_KEY = "rl:",
    
    @Override
    public boolean preHandle(HttpServletRequest request,HttpServletResponse response,Object handler) throws IOException{
        // 请求是否是方法的请求
        if (handler instanceof HandlerMethod){
            HandlerMethod handlerMethod =(HandlerMethod) handler:
            // 获得注解
            RateLimit methodAnnotation = handlerMethod.getMethodAnnotation(RateLimit.class);
            if (methodAnnotation == null) {
                return true;
            }
            int seconds = methodAnnotation.seconds();
            int max = methodAnnotation.max();
            
            String key = RATE_LIMIT_KEY + request.getRemoteAddr()+ ":"+ request.getRequestURI();
            String times = redisTemplate.opsForValue().get(key);
            if (times = null) {
                redisTemplate.opsForValue().set(key, String.value0f(1), seconds, TimeUnit.SECONDS);
            }else{
                int timesInt = Integer.parseInt(times);
                if(timesInt < max){
                    redisTemplate.opsForValue().set(key, String.value0f(timesInt + 1), seconds, TimeUnit.SECONDs);
                }else{
                    response.sendError(429,"to many requests");
                    return false;
                }
            }
        }
        return true,
    }
}

注册到拦截器上

/**
* @author shigenfu
* @date 2023/9/6 11:30 上午
*/

@Configuration
public class WebConfig implements WebMvcConfigurer {
    
    @Resource
    private RateLimitIntercept rateLimitIntercept;
    
    @Override
    public void addInterceptors(InterceptorRegistry registry){
        registry.addInterceptor(rateLimitIntercept),
    }
}

这一步很重要，不然我们自定义的拦截器会不生效。

测试类测试

/**
 * @author shigenfu
 * @date 2023/9/6 11:25 上午
 */
 
@RestController
@RequestMapping(value ="rate")
public class RateLimitTestController {
    
    @GetMapping(value = "test")
    @RateLimit(seconds=10，max=2)
    public String test(){
        return UUID.randomUUID().toString();
    }
}

这里主要是写了一个接口实现测试，访问的时候返回随机生成的UUID字符串。为了测试的效果明显，我在@RateLimit上配置的是10s之内只能允许2次请求。那现在开始测试吧！

测试

依旧用到的是ab这个工具。

 ab -c 2 -n 10 127.0.0.1:9000/rate/test

一起看看效果吧。

可以看到测试的报告，10个请求，8个失败，2个成功，数据库也存储了对应的key和value，很符合预期的效果。

页面的错误是这样的，也很符合预期。

到此这篇关于详解Redis如何优雅地实现接口防刷的文章就介绍到这了,更多相关Redis接口防刷内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

相关文章

  • Redis之常用数据结构哈希表

    Redis之常用数据结构哈希表

    这篇文章主要介绍了Redis常用的数据结构哈希表，哈希表是一种保存键值对的数据结构，具有一定的参考价值，需要的朋友可以参考阅读
    2023-04-04
  • 同一份数据Redis为什么要存两次

    同一份数据Redis为什么要存两次

    这篇文章主要介绍了同一份数据Redis为什么要存两次,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
    2021-01-01
  • Redis的数据过期策略和数据淘汰策略

    Redis的数据过期策略和数据淘汰策略

    本文主要介绍了Redis的数据过期策略和数据淘汰策略,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2025-02-02
  • 详解缓存穿透击穿雪崩解决方案

    详解缓存穿透击穿雪崩解决方案

    在我们日常的开发中，有时需要系统在极短的时间内完成成千上万次的读/写操作，这个时候不是数据库能够承受的，通常会引入NoSQL技术。redis技术就是NoSQL技术中的一种，但是引入redis又有可能出现缓存穿透，缓存击穿，缓存雪崩等问题。本文就对这三种问题进行较深入剖析。
    2021-05-05
  • Redis高并发问题的解决方法

    Redis高并发问题的解决方法

    这篇文章主要介绍了Redis高并发问题的解决办法，具有很好的参考价值，感兴趣的小伙伴们可以参考一下，具体如下：
    2018-05-05
  • Redis数据导入导出以及数据迁移的4种方法详解

    Redis数据导入导出以及数据迁移的4种方法详解

    这篇文章主要介绍了Redis数据导入导出以及数据迁移的4种方法详解,需要的朋友可以参考下
    2020-02-02
  • CentOS8.4安装Redis6.2.6的详细过程

    CentOS8.4安装Redis6.2.6的详细过程

    本文给大家介绍CentOS8.4安装Redis6.2.6的详细过程，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友参考下吧
    2021-11-11
  • 详解Redis数据结构之跳跃表

    详解Redis数据结构之跳跃表

    这篇文章主要介绍了Redis数据结构中的跳跃表的相关知识，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-11-11
  • redis哈希类型_动力节点Java学院整理

    redis哈希类型_动力节点Java学院整理

    这篇文章主要介绍了redis哈希类型的常用方法及原理浅析，感兴趣的朋友一起看看吧
    2017-08-08
  • 关于redis状态监控和性能调优详解

    关于redis状态监控和性能调优详解

    Redis是一种高级key-value数据库。它跟memcached类似，不过数据可以持久化，而且支持的数据类型很丰富。有字符串，链表、哈希、集合和有序集合5种。下面这篇文章主要给大家介绍了关于redis状态监控和性能调优的相关资料，需要的朋友可以参考下。
    2017-09-09

最新评论