springboot 接口限流

在Spring Boot中实现接口限流可以使用Guava RateLimiter。

Guava RateLimiter是一个基于令牌桶算法的限流器，它可以控制在一段时间内某个接口的请求次数，以保护系统免受过多的负载压力。

下面是一个简单的示例，演示如何使用Guava RateLimiter实现接口限流：

1. 添加Guava依赖

在pom.xml文件中添加Guava依赖：

<dependency>
    <groupId>com.google.guava</groupId>
    <artifactId>guava</artifactId>
    <version>30.1.1-jre</version>
</dependency>

2. 创建限流器

在Spring Boot中创建限流器的最简单方法是使用@PostConstruct注解，将限流器初始化在应用程序启动时：

import com.google.common.util.concurrent.RateLimiter;

import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

@Component
public class ApiLimiter {

    private final RateLimiter limiter = RateLimiter.create(10.0); // 每秒最多处理10个请求

    @PostConstruct
    public void init() {
        System.out.println("ApiLimiter initialized.");
    }

    public boolean tryAcquire() {
        return limiter.tryAcquire();
    }
}

3. 在接口方法上添加限流

在需要限流的接口方法上，注入ApiLimiter并调用tryAcquire()方法，如果返回true，则表示可以继续处理请求；如果返回false，则表示请求被限流：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class ApiController {

    @Autowired
    private ApiLimiter apiLimiter;

    @GetMapping("/api")
    public String api() {
        if (apiLimiter.tryAcquire()) {
            // 处理请求
            return "OK";
        } else {
            // 请求被限流
            return "Too many requests.";
        }
    }
}

以上就是在Spring Boot中实现接口限流的简单示例。需要注意的是，Guava RateLimiter只能控制在一段时间内请求的数量，但不能限制每个用户的请求次数。如果需要限制每个用户的请求次数，可以考虑使用Redis等缓存工具实现。