SpringCloud微服务接口限流的两种方式，你学会了吗？

最新推荐文章于 2024-06-28 17:48:41 发布

2401_84048621

最新推荐文章于 2024-06-28 17:48:41 发布

阅读量635

点赞数 11

分类专栏：程序员文章标签： spring cloud 微服务 lua

本文链接：https://blog.csdn.net/2401_84048621/article/details/137868429

版权

程序员专栏收录该内容

451 篇文章 2 订阅

订阅专栏

* 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状态码，然后返回信息到前台展示。