import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.integration.redis.util.RedisLockRegistry;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.concurrent.TimeUnit;
@Configuration
public class RateLimitConfig {
@Bean
public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory redisConnectionFactory) {
RedisTemplate<String, Object> template = new RedisTemplate<>();
template.setConnectionFactory(redisConnectionFactory);
return template;
}
@Bean
public RedisLockRegistry redisLockRegistry(RedisConnectionFactory redisConnectionFactory) {
return new RedisLockRegistry(redisConnectionFactory, "rate-limiting-locks");
}
@Bean
public HandlerInterceptorAdapter rateLimitingInterceptor(RedisLockRegistry lockRegistry) {
return new HandlerInterceptorAdapter() {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws IOException {
String key = request.getRequestURI();
try {
// 尝试获取锁，超时时间设置为1秒
var lock = lockRegistry.obtain(key);
if (lock.tryLock(1, TimeUnit.SECONDS)) {
// 如果获取锁成功，则更新访问计数器
// ...更新访问计数器的逻辑...
return true;
} else {
// 如果未能获取锁，则表示请求过多
response.sendError(HttpServletResponse.SC_TOO_MANY_REQUESTS);
return false;
}
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
return false;
}
}
};
}
}这个代码实例展示了如何在Spring Boot应用中