【SpringBoot应用篇】【AOP+注解】SpringBoot+Guava基于注解实现接口限流+Redis实现分布式接口限流

import com.google.common.annotations.VisibleForTesting;
import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
 
import java.util.concurrent.TimeUnit;
 
@Configuration
public class RateLimitConfiguration {
 
    @Bean
    public RateLimiter rateLimiter() {
        return new RateLimiter();
    }
 
    public static class RateLimiter {
 
        private LoadingCache<String, Long> rateLimiterCache = CacheBuilder.newBuilder()
                .expireAfterWrite(1, TimeUnit.MINUTES) // 缓存有效期1分钟
                .build(new CacheLoader<String, Long>() {
                    // 默认的令牌数
                    @Override
                    public Long load(String key) throws Exception {
                        return 100L; // 假设的默认限流值
                    }
                });
 
        public boolean isRateLimited(String key) {
            try {
                // 尝试消耗令牌
                return !rateLimiterCache.get(key).equals(rateLimiterCache.get(key));
            } catch (Exception e) {
                // 如果发生异常，则允许通过
                return false;
            }
        }
 
        @VisibleForTesting
        void setRate(String key, Long rate) {
            rateLimiterCache.put(key, rate);
        }
    }
}

这个代码示例展示了如何在SpringBoot应用中使用Guava Cache来实现一个简单的限流器。RateLimiter类中的rateLimiterCache是一个缓存操作，其中存储了特定key的令牌数。isRateLimited方法尝试消耗令牌来判断是否应该限流。setRate方法用于测试设置特定key的令牌数。这个简单的例子展示了如何在实际应用中使用注解和AOP来实现接口限流。