使用漏桶算法在 Spring Boot 中进行请求限流

要在 Spring Boot 应用程序中使用漏桶算法对所有用户的请求进行管理，您需要添加一个限流的中间件来处理请求。

下面是修改后的代码示例：

添加依赖项：在项目的 pom.xml 文件中添加以下依赖项。

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

    <!-- Google Guava -->
    <dependency>
        <groupId>com.google.guava</groupId>
        <artifactId>guava</artifactId>
        <version>30.1-jre</version>
    </dependency>
</dependencies>

创建一个限流中间件类，用于实现漏桶算法。在该类中添加代码来处理请求并进行限流。以下是一个简单的示例：

import com.google.common.util.concurrent.RateLimiter;
import org.springframework.stereotype.Component;

import javax.servlet.*;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

@Component
public class RateLimitFilter implements Filter {

    private static final int MAX_REQUESTS_PER_SECOND = 10; // 每秒最大请求数
    private static final RateLimiter rateLimiter = RateLimiter.create(MAX_REQUESTS_PER_SECOND);

    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
        HttpServletResponse httpResponse = (HttpServletResponse) response;

        if (rateLimiter.tryAcquire()) {
            // 请求未被限流，继续处理
            chain.doFilter(request, response);
        } else {
            // 请求被限流，返回错误响应
            httpResponse.setStatus(HttpServletResponse.SC_TOO_MANY_REQUESTS);
            httpResponse.getWriter().write('Too many requests. Please try again later.');
        }
    }
}

更新控制器类，添加 @WebFilter 注解，将限流中间件应用到所有请求上。

import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("/dorm")
public class DormSelectionController {

    @GetMapping("/select")
    public String selectDorm() {
        // 处理宿舍选择逻辑
        return "Dorm selected!";
    }

    @Bean
    public FilterRegistrationBean<RateLimitFilter> rateLimitFilter() {
        FilterRegistrationBean<RateLimitFilter> registrationBean = new FilterRegistrationBean<>();
        registrationBean.setFilter(new RateLimitFilter());
        registrationBean.addUrlPatterns("/dorm/*"); // 设置需要限流的路径
        return registrationBean;
    }
}

运行应用程序，并在浏览器中访问 http://localhost:8080/dorm/select。根据漏桶算法的限流策略，每秒最多只能处理 10 个请求。如果超过限制，将返回状态码 429（太多请求）和错误消息 'Too many requests. Please try again later.'。

通过以上步骤，您已经成功实现了使用漏桶算法进行请求限流的功能。请根据您的需求进行修改和扩展。希望对您有所帮助！