Java Interceptor 拦截器如何实现请求限流

在Java中，拦截器（Interceptor）通常用于在方法调用前后执行一些额外的逻辑。请求限流（Rate Limiting）是一种控制请求速率的技术，以防止系统过载。结合这两者，我们可以在拦截器中实现请求限流。

以下是一个使用Spring框架的拦截器实现请求限流的示例：

1. 首先，添加依赖：

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

2. 创建一个限流拦截器：

import org.springframework.web.servlet.HandlerInterceptor;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;

public class RateLimitInterceptor implements HandlerInterceptor {

    private final int limit = 5; // 每秒允许的最大请求次数
    private final ConcurrentHashMap<String, AtomicInteger> requestCountMap = new ConcurrentHashMap<>();

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String clientIP = request.getRemoteAddr();
        AtomicInteger requestCount = requestCountMap.computeIfAbsent(clientIP, k -> new AtomicInteger(0));

        if (requestCount.incrementAndGet() > limit) {
            response.setStatus(HttpServletResponse.SC_TOO_MANY_REQUESTS);
            response.getWriter().write("请求过于频繁，请稍后再试");
            return false;
        }

        // 设置1秒后请求计数减1
        new Thread(() -> {
            try {
                Thread.sleep(1000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            requestCount.decrementAndGet();
        }).start();

        return true;
    }
}

3. 注册拦截器：

import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

@Configuration
public class WebConfig implements WebMvcConfigurer {

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new RateLimitInterceptor()).addPathPatterns("/**");
    }
}

这个示例中，我们创建了一个名为RateLimitInterceptor的拦截器，它限制每个客户端IP每秒最多只能发送5个请求。如果超过限制，将返回HTTP 429状态码（Too Many Requests）。

在preHandle方法中，我们使用ConcurrentHashMap来存储每个客户端的请求计数。当请求到达时，我们增加对应客户端的请求计数，并在一个新线程中设置1秒后请求计数减1。如果请求计数超过限制，我们返回HTTP 429状态码并拒绝请求。

最后，我们在WebConfig类中注册拦截器，并为其添加一个路径模式（这里使用/**表示拦截所有请求）。