快捷導(dǎo)航

SpringBoot使用Redis進(jìn)行限流功能實現(xiàn)

更新時間：2025年10月23日 09:38:29 作者：小猿、

Spring Boot中使用Redis實現(xiàn)限流功能是一種常見的做法,特別是在高并發(fā)場景下,限流可以有效防止系統(tǒng)過載,保證服務(wù)的穩(wěn)定性,本文就來詳細(xì)的介紹一下SpringBoot使用Redis進(jìn)行限流功能實現(xiàn),感興趣的可以了解一下

概述

Spring Boot中使用Redis實現(xiàn)限流功能是一種常見的做法，特別是在高并發(fā)場景下，限流可以有效防止系統(tǒng)過載，保證服務(wù)的穩(wěn)定性。

1. 限流的應(yīng)用場景

限流通常用于以下場景：

API限流：防止API被惡意或過度調(diào)用，保護(hù)后端服務(wù)。
防止爬蟲：限制爬蟲的訪問頻率，防止數(shù)據(jù)被大量抓取。
秒殺系統(tǒng)：在秒殺活動中，限制用戶的請求頻率，防止系統(tǒng)崩潰。
防止DDoS攻擊：通過限流防止分布式拒絕服務(wù)攻擊。

2. Redis限流的實現(xiàn)原理

Redis限流的常見實現(xiàn)方式有：

計數(shù)器算法：通過Redis的INCR命令實現(xiàn)簡單的計數(shù)器限流。
滑動窗口算法：通過Redis的有序集合（ZSET）實現(xiàn)滑動窗口限流。
令牌桶算法：通過Redis的LIST或ZSET實現(xiàn)令牌桶限流。

下面我們以計數(shù)器算法為例，講解如何在Spring Boot中實現(xiàn)限流。

3. Spring Boot集成Redis實現(xiàn)限流

3.1 添加依賴

首先，在pom.xml中添加Spring Boot和Redis的依賴：

<dependencies>
    <!-- Spring Boot Starter -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>

    <!-- Spring Boot Redis Starter -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-data-redis</artifactId>
    </dependency>

    <!-- Lettuce Redis Client -->
    <dependency>
        <groupId>io.lettuce</groupId>
        <artifactId>lettuce-core</artifactId>
    </dependency>
</dependencies>

3.2 配置Redis

在application.properties中配置Redis連接信息：

spring.redis.host=localhost
spring.redis.port=6379

3.3 實現(xiàn)限流邏輯

我們可以通過自定義注解和AOP來實現(xiàn)限流功能。

3.3.1 自定義限流注解

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

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface RateLimiter {
    String key(); // Redis的key
    int limit();  // 限流次數(shù)
    int expire(); // 過期時間（秒）
}

3.3.2 實現(xiàn)AOP切面

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

@Aspect
@Component
public class RateLimiterAspect {

    @Autowired
    private RedisTemplate<String, Object> redisTemplate;

    @Around("@annotation(rateLimiter)")
    public Object around(ProceedingJoinPoint joinPoint, RateLimiter rateLimiter) throws Throwable {
        String key = rateLimiter.key();
        int limit = rateLimiter.limit();
        int expire = rateLimiter.expire();

        // 獲取當(dāng)前請求的次數(shù)
        Long count = redisTemplate.opsForValue().increment(key, 1);

        if (count == 1) {
            // 如果是第一次請求，設(shè)置過期時間
            redisTemplate.expire(key, expire, TimeUnit.SECONDS);
        }

        if (count > limit) {
            // 超過限流次數(shù)，拋出異?；蚍祷劐e誤信息
            throw new RuntimeException("請求過于頻繁，請稍后再試");
        }

        // 執(zhí)行目標(biāo)方法
        return joinPoint.proceed();
    }
}

3.3.3 使用限流注解

在Controller中使用@RateLimiter注解：

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("/api")
public class ApiController {

    @RateLimiter(key = "api:limit:test", limit = 5, expire = 60)
    @GetMapping("/test")
    public String test() {
        return "請求成功";
    }
}

4. 示例代碼解釋

@RateLimiter注解：用于標(biāo)記需要限流的方法，指定限流的key、限制次數(shù)和過期時間。
RateLimiterAspect切面：通過AOP攔截帶有@RateLimiter注解的方法，使用Redis的INCR命令實現(xiàn)計數(shù)器限流。
RedisTemplate：用于操作Redis，實現(xiàn)計數(shù)器的自增和過期時間設(shè)置。