Guava之RateLimiter限流

最新推荐文章于 2024-08-17 15:35:11 发布

BugRoot

最新推荐文章于 2024-08-17 15:35:11 发布

阅读量3.1k

点赞数

分类专栏： Java 优化组件

本文链接：https://blog.csdn.net/BugRoot/article/details/116718248

版权

本文介绍了Guava的RateLimiter如何应用于限流，以防止服务器因高频访问而宕机。讲解了RateLimiter在物流系统查询接口保护中的作用，以及令牌算法的原理。通过自定义注解、切面处理实现限流，并解析了RateLimiter的相关参数和接口，包括`expireAfterWrite`的含义。最后提到了分布式Redis限流的后续讨论。

摘要由CSDN通过智能技术生成

RateLimter是什么，我们为什么需要用到它，以物流系统作为例子：比如系统有一个物流信息查询接口，提供给第三方调用，接口暴露在公网，会出现什么问题，大致讲下如下问题：
1、大量正常用户高频访问导致服务器宕机
2、恶意用户高频访问导致服务器宕机
3、网络爬虫等等

限流常见方案：
1：Nginx限流
2：Tomcat限流
3：漏桶算法
4：令牌算法等等

这里不多介绍，只介绍下RateLimter的令牌算法。
其它的后续文章会加上。
需要引入第三方组件库

<dependency>
     <groupId>com.google.guava</groupId>
     <artifactId>guava</artifactId>
     <version>25.1-jre</version>
</dependency>

在这里插入图片描述
这边通过切面的方式
首先我们需要自定义一个注解

import java.lang.annotation.*;

/**
 * 自定义注解  限流
 */
@Documented
@Target({
   ElementType.PARAMETER, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface RLimit {
   

    /**
     *  key
     */
    String key() default "";
    /**
     *  备注
     */
    String description()  default "";
    /**
     *  限流的类型
     */
    LimitType limitType() default LimitType.RANDOM_KEY;

    enum LimitType {
   
        /**
         * 自定义key
         */
        RANDOM_KEY,
        /**
         * 根据请求者IP
         */
        IP
    }
}

切面类
在这里插入图片描述

我们分析下代码
import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import com.google.common.util.concurrent.RateLimiter;
import com.peng.sentinel.GaciException;
import com.peng.sentinel.annotation.RLimit;
import com.peng.sentinel.util.IPHelper;
import org.aspectj.lang.ProceedingJoinPoint;</