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来实现接口限流。