Spring Cloud Gateway基于CPU使用率实现限流

胖虎

发布于 2019-06-26 17:12:33

1.1K0

发布于 2019-06-26 17:12:33

文章被收录于专栏：晏霖

前言

本文Demo摘自《重新定义》

在项目里利用 cpu 的使用率来做限流其实不是很常用，或者基本不会使用这种方法进行，限流，除非该项目有他特殊的需要，因为我们不能保证是其他应用或者服务引起的 cpu 使用率增加，还是由于访问量大，导致 cpu 使用率增加，因为一台机器 cpu使用率不是说你一个 java服务就能控制的，所以用这个判断是有误差的，本文就是一个为了扩展大家思路而写的。

正文

思想就是我们通过观察机器 cpu使用率，然后设定一个阀值，超过阀值不允许访问。

代码清单

@Componentpublic class GatewayRateLimitFilterByCpu implements GatewayFilter, Ordered {
    private final Logger log = LoggerFactory.getLogger(GatewayRateLimitFilterByCpu.class);
    @Autowired    private MetricsEndpoint metricsEndpoint;
    private static final String METRIC_NAME = "system.cpu.usage";
    private static final double MAX_USAGE = 0.050D;
    @Override    public Mono<Void> filter(ServerWebExchange exchange, 
GatewayFilterChain chain) {
        //获取网关所在机器的CPU使用情况        Double systemCpuUsage = metricsEndpoint.metric(METRIC_NAME, null)                .getMeasurements()                .stream()                .filter(Objects::nonNull)                .findFirst()                .map(MetricsEndpoint.Sample::getValue)                .filter(Double::isFinite)                .orElse(0.0D);
        boolean isOpenRateLimit = systemCpuUsage >MAX_USAGE;        log.debug("system.cpu.usage: {}, isOpenRateLimit:{} ",systemCpuUsage , isOpenRateLimit);        if (isOpenRateLimit) {            //当CPU的使用超过设置的最大阀值开启限流            exchange.getResponse().setStatusCode(HttpStatus.TOO_MANY_REQUESTS);            return exchange.getResponse().setComplete();        } else {            return chain.filter(exchange);        }    }
    @Override    public int getOrder() {        return 0;    }
}

我们通过 api 方式配置过滤器作用到某个路由，其实我们也可以在yml 文件中配置路由地址的映射。代码如下

@SpringBootApplication
public class GatewayApplication {

    @Autowired
    private GatewayRateLimitFilterByCpu gatewayRateLimitFilterByCpu;

    @Bean
    public RouteLocator customerRouteLocator(RouteLocatorBuilder builder) {
        return builder.routes()
                .route(r -> r.path("/test/rateLimit")
                        .filters(f -> f.filter(gatewayRateLimitFilterByCpu))
                        .uri("http://localhost:8000/hello/rateLimit")
                        .id("rateLimit_route")
                ).build();
    }
    public static void main(String[] args) {
        SpringApplication.run(GatewayApplication.class, args);
    }
}