百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术教程 > 正文

接口也会累着!Spring Boot + Redis 实现接口限流和防刷

mhr18 2024-11-24 18:45 27 浏览 0 评论

越来越多的应用被部署到了云端,并在云端被高并发的访问。在这样的环境下,限流和防刷是非常重要的,因为如果不做好限流和防刷,那么很容易就会因为访问量过大而导致应用崩溃。在本文中,我们将会介绍如何使用Spring Boot和Redis来实现访问限流和接口防刷。篇幅有点长,但是满满的干活,大家地铁上慢慢看!!!

什么是限流和防刷?

在介绍如何使用Spring Boot和Redis实现限流和防刷之前,我们先来了解一下什么是限流和防刷。

限流,顾名思义,就是限制流量,也就是限制同一时间内能够访问某个接口的请求数量。通过限流,我们可以保证系统的稳定性,避免因为访问量过大而导致系统宕机的情况发生。

防刷,是指防止恶意请求,比如恶意爬虫、恶意攻击等。如果系统没有做好防刷,那么恶意请求很容易就会把系统压垮,因此防刷也是非常重要的一项工作。

Spring Boot和Redis

在介绍如何使用Spring Boot和Redis实现限流和防刷之前,我们先来了解一下什么是Spring Boot和Redis。

Spring Boot是一个开源的Java框架,它可以帮助我们快速搭建基于Spring的应用程序,减少了我们的开发时间和成本,同时提高了应用程序的可维护性。

Redis是一个高性能的键值存储系统,它支持多种数据结构,比如字符串、哈希表、列表等等。Redis可以用于缓存、消息队列、计数器等多种用途。

实现限流

接下来,我们将会介绍如何使用Spring Boot和Redis来实现限流。

在Redis中,我们可以使用令牌桶算法来实现限流。令牌桶算法的基本思想是,我们有一个固定容量的桶,以固定的速率往里面放入令牌,每当有一个请求需要访问系统时,就需要从桶中取出一个令牌,如果桶中没有令牌了,那么就需要等待,直到桶中有令牌为止。

首先,我们需要在pom.xml中添加以下依赖:

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>

接下来,我们需要在Redis中设置一个哈希表,用来存储每个接口的访问次数和访问时间。其中,哈希表的键是接口的路径,值是一个列表,列表中存储了接口的访问次数和访问时间。我们可以使用以下代码来初始化这个哈希表:

@Autowired
private RedisTemplate<String, Object> redisTemplate;

private static final String LIMIT_KEY_PREFIX = "limit:";

public void initLimitMap(String path, int limitCount, int limitTime) {
    String key = LIMIT_KEY_PREFIX + path;
    Map<String, Integer> map = new HashMap<>();
    map.put("count", 0);
    map.put("time", limitTime);
    redisTemplate.opsForHash().putAll(key, map);
    redisTemplate.expire(key, limitTime, TimeUnit.SECONDS);
}

在上面的代码中,我们使用了@Autowired注解来注入RedisTemplate实例。RedisTemplate是Spring提供的一个用于操作Redis的模板类,可以简化我们的Redis操作。

然后,我们可以使用initLimitMap方法来初始化每个接口的访问次数和访问时间。其中,path是接口的路径,limitCount是访问次数的限制,limitTime是访问时间的限制。

接下来,我们需要编写一个拦截器,在每个请求到达接口之前判断是否需要进行限流。在拦截器中,我们需要从Redis中获取当前接口的访问次数和访问时间,然后判断当前时间和上次访问时间的时间差是否超过访问时间的限制,如果超过了,则重置访问次数为1,否则增加访问次数。如果当前访问次数超过了访问次数的限制,则拒绝访问。

以下是一个简单的限流拦截器的代码:

public class LimitInterceptor extends HandlerInterceptorAdapter {
    
    @Autowired
    private RedisTemplate<String, Object> redisTemplate;

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String path = request.getRequestURI();
        String key = LIMIT_KEY_PREFIX + path;
        BoundHashOperations<String, String, Integer> boundHashOps = redisTemplate.boundHashOps(key);

        int count = boundHashOps.get("count");
        int time = boundHashOps.get("time");
        long currentTime = System.currentTimeMillis();
        long lastTime = boundHashOps.get("lastTime");
        if (lastTime == 0) {
            lastTime = currentTime;
        }
        if (currentTime - lastTime > time * 1000) {
            count = 1;
        } else {
            count++;
        }
        boundHashOps.put("count", count);
        boundHashOps.put("lastTime", (int) currentTime);

        if (count > limitCount) {
            response.setContentType("application/json;charset=utf-8");
            response.getWriter().write("请求太频繁");
            return false;
        }

        return true;
    }
}

在上面的代码中,我们使用了BoundHashOperations类来操作Redis中的哈希表。BoundHashOperations是RedisTemplate的一个子类,可以更加方便地操作Redis中的哈希表。

我们在拦截器中,首先获取当前接口的路径,并根据路径获取当前接口的限流信息。然后,我们判断当前时间和上次访问时间的时间差是否超过访问时间的限制,如果超过了,则重置访问次数为1,否则增加访问次数。最后,如果当前访问次数超过了访问次数的限制,则拒绝访问。

最后,我们需要在Spring Boot中注册我们的限流拦截器。我们可以使用以下代码来注册拦截器:

@Configuration
public class WebConfig implements WebMvcConfigurer {
    
    @Autowired
    private LimitInterceptor limitInterceptor;

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(limitInterceptor);
    }
}

在上面的代码中,我们创建了一个WebConfig类,并使用@Configuration注解来声明该类是一个配置类。然后,我们实现了WebMvcConfigurer接口,并重写了其中的addInterceptors方法。在该方法中,我们将我们的限流拦截器注册到了拦截器列表中。

现在,我们可以使用上面的代码来实现接口的限流了。当有用户请求某个接口时,我们会根据用户的IP地址和接口的路径来判断是否需要进行限流,如果需要进行限流,则根据接口的限流配置来判断是否拒绝请求。同时,我们还可以使用Redis的持久化功能来保证我们的限流信息不会因为服务器重启而丢失。

在实际应用中,我们还可以根据业务需求来调整接口的限流策略。例如,对于一些特殊的接口,我们可以将限流次数设置得更高,以保证这些接口能够稳定运行。同时,我们还可以使用一些工具来监控接口的访问情况,以便及时调整限流策略。

总结

本文介绍了如何使用Spring Boot和Redis来实现接口的限流和防刷。通过使用Redis来存储限流信息,我们可以在分布式环境中方便地实现限流功能。同时,我们还可以根据业务需求来调整限流策略,以提高接口的可用性和稳定性。

需要注意的是,接口的限流和防刷只是保证接口能够正常运行的一种手段,我们还需要在代码中加入其他的安全措施,例如输入参数的校验、防SQL注入、防XSS攻击等。只有综合使用各种安全手段,才能有效地保障系统的安全性和稳定性。

下一篇内容我们来讨论一下接口防刷!

相关推荐

【推荐】一个开源免费、AI 驱动的智能数据管理系统,支持多数据库

如果您对源码&技术感兴趣,请点赞+收藏+转发+关注,大家的支持是我分享最大的动力!!!.前言在当今数据驱动的时代,高效、智能地管理数据已成为企业和个人不可或缺的能力。为了满足这一需求,我们推出了这款开...

Pure Storage推出统一数据管理云平台及新闪存阵列

PureStorage公司今日推出企业数据云(EnterpriseDataCloud),称其为组织在混合环境中存储、管理和使用数据方式的全面架构升级。该公司表示,EDC使组织能够在本地、云端和混...

对Java学习的10条建议(对java课程的建议)

不少Java的初学者一开始都是信心满满准备迎接挑战,但是经过一段时间的学习之后,多少都会碰到各种挫败,以下北风网就总结一些对于初学者非常有用的建议,希望能够给他们解决现实中的问题。Java编程的准备:...

SQLShift 重大更新:Oracle→PostgreSQL 存储过程转换功能上线!

官网:https://sqlshift.cn/6月,SQLShift迎来重大版本更新!作为国内首个支持Oracle->OceanBase存储过程智能转换的工具,SQLShift在过去一...

JDK21有没有什么稳定、简单又强势的特性?

佳未阿里云开发者2025年03月05日08:30浙江阿里妹导读这篇文章主要介绍了Java虚拟线程的发展及其在AJDK中的实现和优化。阅前声明:本文介绍的内容基于AJDK21.0.5[1]以及以上...

「松勤软件测试」网站总出现404 bug?总结8个原因,不信解决不了

在进行网站测试的时候,有没有碰到过网站崩溃,打不开,出现404错误等各种现象,如果你碰到了,那么恭喜你,你的网站出问题了,是什么原因导致网站出问题呢,根据松勤软件测试的总结如下:01数据库中的表空间不...

Java面试题及答案最全总结(2025版)

大家好,我是Java面试陪考员最近很多小伙伴在忙着找工作,给大家整理了一份非常全面的Java面试题及答案。涉及的内容非常全面,包含:Spring、MySQL、JVM、Redis、Linux、Sprin...

数据库日常运维工作内容(数据库日常运维 工作内容)

#数据库日常运维工作包括哪些内容?#数据库日常运维工作是一个涵盖多个层面的综合性任务,以下是详细的分类和内容说明:一、数据库运维核心工作监控与告警性能监控:实时监控CPU、内存、I/O、连接数、锁等待...

分布式之系统底层原理(上)(底层分布式技术)

作者:allanpan,腾讯IEG高级后台工程师导言分布式事务是分布式系统必不可少的组成部分,基本上只要实现一个分布式系统就逃不开对分布式事务的支持。本文从分布式事务这个概念切入,尝试对分布式事务...

oracle 死锁了怎么办?kill 进程 直接上干货

1、查看死锁是否存在selectusername,lockwait,status,machine,programfromv$sessionwheresidin(selectsession...

SpringBoot 各种分页查询方式详解(全网最全)

一、分页查询基础概念与原理1.1什么是分页查询分页查询是指将大量数据分割成多个小块(页)进行展示的技术,它是现代Web应用中必不可少的功能。想象一下你去图书馆找书,如果所有书都堆在一张桌子上,你很难...

《战场兄弟》全事件攻略 一般事件合同事件红装及隐藏职业攻略

《战场兄弟》全事件攻略,一般事件合同事件红装及隐藏职业攻略。《战场兄弟》事件奖励,事件条件。《战场兄弟》是OverhypeStudios制作发行的一款由xcom和桌游为灵感来源,以中世纪、低魔奇幻为...

LoadRunner(loadrunner录制不到脚本)

一、核心组件与工作流程LoadRunner性能测试工具-并发测试-正版软件下载-使用教程-价格-官方代理商的架构围绕三大核心组件构建,形成完整测试闭环:VirtualUserGenerator(...

Redis数据类型介绍(redis 数据类型)

介绍Redis支持五种数据类型:String(字符串),Hash(哈希),List(列表),Set(集合)及Zset(sortedset:有序集合)。1、字符串类型概述1.1、数据类型Redis支持...

RMAN备份监控及优化总结(rman备份原理)

今天主要介绍一下如何对RMAN备份监控及优化,这里就不讲rman备份的一些原理了,仅供参考。一、监控RMAN备份1、确定备份源与备份设备的最大速度从磁盘读的速度和磁带写的带度、备份的速度不可能超出这两...

取消回复欢迎 发表评论: