2024-02-20｜RateLimitAspect请求限流、调整spring-cloud-examples项目结构

   @Aspect
   @Component
   public class RateLimitAspect {
     private final RedisTemplate<String, Object> redisTemplate;
     private final Logger logger = LoggerFactory.getLogger(RateLimitAspect.class);
   
     private final int maxRequests; // Maximum number of requests
     private final int timeWindow; // Time window in seconds
   
     @Autowired
     public RateLimitAspect(RedisTemplate<String, Object> redisTemplate) {
         this.redisTemplate = redisTemplate;
         this.maxRequests = 100; // Default maximum number of requests is 100
         this.timeWindow = 60; // Default time window is 60 seconds
     }
   
     @Before("@annotation(getMapping)")
     public void applyRateLimit(JoinPoint joinPoint, GetMapping getMapping) {
         HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder
            .getRequestAttributes()).getRequest();
         if (request != null && HttpMethod.GET.matches(request.getMethod())) {
             String username = request.getUserPrincipal().getName(); // Get the username
             if (username != null && !username.isEmpty()) {
                 String rateLimitKey = "rate_limit:" + username;
                 incrementAndCheckRateLimit(rateLimitKey, request);
             }
         }
     }
   
     private void incrementAndCheckRateLimit(String rateLimitKey, HttpServletRequest request) {
         Long currentRequests = redisTemplate.execute((RedisCallback<Long>) connection -> {
             Object nativeConnection = connection.getNativeConnection();
             String script = "local current = redis.call('INCR', KEYS[1])\n"
                     + "if tonumber(current) == 1 then\n"
                     + "    redis.call('EXPIRE', KEYS[1], ARGV[1])\n"
                     + "end\n"
                     + "return current";
             return (Long) ((RedisOperations<?, ?>) nativeConnection).execute(
                     (RedisCallback<Object>) connection1 -> connection1.eval(
                             script.getBytes(),
                             redisTemplate.getKeySerializer(), // Use custom Key serializer
                             redisTemplate.getValueSerializer(), // Use custom Value serializer
                             Collections.singletonList(rateLimitKey.getBytes()),
                             Collections.singletonList(String.valueOf(timeWindow).getBytes())
                     )
             );
         });
   
         if (currentRequests > maxRequests) {
             String urlWithParams = getRequestUrlWithParams(request);
             logger.warn("Rate limit exceeded for key: {}. Request URL with Params: {}", 
                rateLimitKey, urlWithParams);
             throw new RateLimitException();
         }
     }
   
     private String getRequestUrlWithParams(HttpServletRequest request) {
         String url = request.getRequestURL().toString();
         String queryString = request.getQueryString();
         if (queryString != null && !queryString.isEmpty()) {
             url += "?" + queryString;
         }
         return url;
     }
   }