* limitations under the License.
*/
package com.marcosbarbero.cloud.autoconfigure.zuul.ratelimit.config.properties;
public enum RateLimitRepository {
/**
* Uses Redis as data storage
*/
REDIS,
/**
* Uses Consul as data storage
*/
CONSUL,
/**
* Uses SQL database as data storage
*/
JPA,
/**
* Uses Bucket4j JCache as data storage
*/
BUCKET4J_JCACHE,
/**
* Uses Bucket4j Hazelcast as data storage
*/
BUCKET4J_HAZELCAST,
/**
* Uses Bucket4j Ignite as data storage
*/
BUCKET4J_IGNITE,
/**
* Uses Bucket4j Infinispan as data storage
*/
BUCKET4J_INFINISPAN,
}
可以看到这是一个枚举类。其中,较为熟悉的就是REDIS、CONSUL、JPA。这里,就选择REDIS作为限流数据的存储。这就是pom文件中导入REDIS的原因。重新修改配置文件
ratelimit:
enabled: true
repository: REDIS #对应存储类型(用来存储统计信息)
behind-proxy: true #代理之后
default-policy: #可选 - 针对所有的路由配置的策略,除非特别配置了policies
limit: 10 #可选 - 每个刷新时间窗口对应的请求数量限制
quota: 4 #可选- 每个刷新时间窗口对应的请求时间限制(秒)
refresh-interval: 5 # 刷新时间窗口的时间,默认值 (秒)
type: #可选 限流方式
- url
服务正常启动,接下来进行测试。学习资料:Java进阶视频资源
上面限制5秒之内只能发送10次请求,那就发送11次请求试一试。使用postman工具进行请求的发送。同时,教大家如何使用postman批量发送请求。
首先,在地址栏填写接口地址,然后,可以点击send,测试当前地址是否正确。如果一切正常,点击旁边的save按钮。
出现上面的浮层,按照箭头的标注,进行操作,点击save按钮后,在左边就能看到已经保存的请求了。
然后点击图示的run按钮,进入下面的界面。
按照文中给出的注释,进行设定,这里我就改一下请求的次数,设置为11次。然后点击run按钮发起请求,看测试结果。
可以看到,配置文件中规定请求次数为10次,测试前10次请求都是正常的响应。而第11次请求就被限流策略阻止了。证明了配置生效,接口限流成功。学习资料:Java进阶视频资源
方法二:过滤器限流
Zuul的过滤器功能,不仅可以做权限验证,也可以拿来做限流。所以,新建一个过滤器,让其继承ZuulFilter
package com.root.project.apigateway.filter;
import com.google.common.util.concurrent.RateLimiter;
import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.exception.ZuulException;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Component;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import static org.springframework.cloud.netflix.zuul.filters.support.FilterConstants.PRE_TYPE;
/**
* @ClassName: InterfaceLimitFilter
* @Author: 清风一阵吹我心
* @Description: TODO
* @Version 1.0
**/
@Component
public class InterfaceLimitFilter extends ZuulFilter {
//每秒产生10个令牌
private static final RateLimiter RATE_LIMITER = RateLimiter.create(10);
@Override
public String filterType() {
return PRE_TYPE;
}
@Override
public int filterOrder() {
return -10;
}
@Override
public boolean shouldFilter() {
RequestContext context = RequestContext.getCurrentContext();
HttpServletRequest request = context.getRequest();
//只对管理员服务限流
String admin = “/api/admin/v1.0/admin”;
return admin.equalsIgnoreCase(request.getRequestURI());
}
/**
* 限流逻辑
* RATE_LIMITER.tryAcquire()尝试获取令牌
* @return
* @throws ZuulException
*/
@Override
public Object run() throws ZuulException {
RequestContext context = RequestContext.getCurrentContext();
//每调用一次tryAcquire()方法,令牌数量减1
if (!RATE_LIMITER.tryAcquire()) {
context.setSendZuulResponse(false);
context.setResponseStatusCode(HttpStatus.TOO_MANY_REQUESTS.value());
HttpServletResponse response = context.getResponse();
try {
response.setContentType(“text/html;charset=UTF-8”);
response.getWriter().write(“访问上限,限流啦”);
} catch (IOException e) {
e.printStackTrace();
}
}
return null;
}
}
因为Guava是基于令牌桶算法的实现类,所以RateLimiter.create(10)
可以理解为:每秒往桶里放入10个令牌。RATE_LIMITER.tryAcquire()
方法尝试获取桶里的令牌,如果有,则返回true,并且,总的令牌数减1。没有则返回false。后面的逻辑和权限验证相似。设置Htpp状态码,然后返回信息到前台展示。
注意:在类上还要加入
@Component
注解,将其注入到IOC容器中。
启动服务,进行测试。使用postman发送20个请求。
奇怪的事情发生了,可以看到第十八个请求被限流了,但是第十九个请求又成功了。其实这也不难理解,当第十八个请求到达网关时,木桶里已经没有令牌可以获取了。所以,直接被限流策略限制。
其实,除了网关层的限流,还能在Nginx层上做限流。感兴趣的朋友,可以去看一下Nginx方面的知识。一个高性能的HTTP和反向代理服务器,作为一名开发人员,了解和使用它,也是必不可少的。
来源:blog.csdn.net/qq_32101993/
article/details/87375917
推荐:
最后
自我介绍一下,小编13年上海交大毕业,曾经在小公司待过,也去过华为、OPPO等大厂,18年进入阿里一直到现在。
深知大多数Java工程师,想要提升技能,往往是自己摸索成长,自己不成体系的自学效果低效漫长且无助。
因此收集整理了一份《2024年Java开发全套学习资料》,初衷也很简单,就是希望能够帮助到想自学提升又不知道该从何学起的朋友,同时减轻大家的负担。
既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上Java开发知识点,不论你是刚入门Android开发的新手,还是希望在技术上不断提升的资深开发者,这些资料都将为你打开新的学习之门!
如果你觉得这些内容对你有帮助,需要这份全套学习资料的朋友可以戳我获取!!
由于文件比较大,这里只是将部分目录截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且会持续更新!
)]
[外链图片转存中…(img-c37G8A2B-1715081736182)]
[外链图片转存中…(img-C2tvr8fN-1715081736183)]
既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上Java开发知识点,不论你是刚入门Android开发的新手,还是希望在技术上不断提升的资深开发者,这些资料都将为你打开新的学习之门!
如果你觉得这些内容对你有帮助,需要这份全套学习资料的朋友可以戳我获取!!
由于文件比较大,这里只是将部分目录截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且会持续更新!