阿里Sentinel整合Zuul网关详解

前面我们讲解了Sentinel整合Spring Cloud Gateway，详细请查看文章：阿里Sentinel支持Spring Cloud Gateway啦

目前来说，大部分公司线上的网关应该是Zuul，所以今天我们就来看看如何在Zuul中整合Sentinel。本来想基于Spring Cloud Alibaba来进行整合讲解，整合的时候发现目前还没更新版本，依赖还是之前的版本，咱们就以最原生的方式进行整合吧，等Spring Cloud Alibaba更新之后，Sentinel的整合只会变得更简单。

加入zuul-adapter依赖：

<dependency>
	<groupId>com.alibaba.csp</groupId>
	<artifactId>sentinel-zuul-adapter</artifactId>
	<version>1.6.0</version>
</dependency>

配置Sentinel提供的限流过滤器和限流规则：

@Configuration
public class ZuulConfig {

	@Bean
	public ZuulFilter sentinelZuulPreFilter() {
		return new SentinelZuulPreFilter();
	}

	@Bean
	public ZuulFilter sentinelZuulPostFilter() {
		return new SentinelZuulPostFilter();
	}

	@Bean
	public ZuulFilter sentinelZuulErrorFilter() {
		return new SentinelZuulErrorFilter();
	}

	@PostConstruct
	public void doInit() {
		// 注册 FallbackProvider
		ZuulBlockFallbackManager.registerProvider(new MyBlockFallbackProvider());
		initGatewayRules();
	}

	/**
	 * 配置限流规则
	 */
	private void initGatewayRules() {
		Set<GatewayFlowRule> rules = new HashSet<>();
		rules.add(new GatewayFlowRule("yinjihuan").setCount(1) // 限流阈值
				.setIntervalSec(1) // 统计时间窗口，单位是秒，默认是 1 秒
		);

		GatewayRuleManager.loadRules(rules);
	}
}

• SentinelZuulPreFilter
  pre过滤器，在请求路由之前匹配routeId和api，进行限流操作
• SentinelZuulPostFilter
  post过滤器，路由之后恢复资源
• SentinelZuulErrorFilter
  error过滤器，异常后的处理

最后再配置一个简单的路由，路由名称yinjihuan，跟上面规则中的名称一致：

zuul.routes.yinjihuan.path=/cxytiandi/**
zuul.routes.yinjihuan.url=http://cxytiandi.com

触发限流后会返回固定的提示：

{
    "code":429,
    "message":"Sentinel block exception",
    "route":"yinjihuan"
}

如果想修改提示内容可以自己实现ZuulBlockFallbackProvider接口，框架默认提供的实现是DefaultBlockFallbackProvider，源码如下：

public class DefaultBlockFallbackProvider implements ZuulBlockFallbackProvider {

    @Override
    public String getRoute() {
        return "*";
    }

    @Override
    public BlockResponse fallbackResponse(String route, Throwable cause) {
        if (cause instanceof BlockException) {
            return new BlockResponse(429, "Sentinel block exception", route);
        } else {
            return new BlockResponse(500, "System Error", route);
        }
    }
}

用法其实跟Zuul中的FallbackProvider一致，但是FallbackProvider比较好的是返回的ClientHttpResponse，我们可以自定义响应内容。

Sentinel提供的ZuulBlockFallbackProvider接口中定义的返回对象是BlockResponse ，也就意味着限制了响应的字段，BlockResponse中有code,message,route三个字段，如果我想返回status, msg这两个字段目前我没找到其它的方式，不知道后续会不会支持，其实最好的是也返回ClientHttpResponse，这样就可以自定义响应内容了。

这边有个小插曲，就是我们自定义fallbackResponse的时候如果用中文message的话，响应内容是乱码，如下：

{
code: 429,
message: "??????",
route: "yinjihuan"
}

我看了下SentinelZuulPreFilter中的代码，如下：

这边是构造了BlockResponse，然后设置到ResponseBody中，但是没有进行编码设置，我自己改了下源码，加了一行代码：

  ctx.getResponse().setContentType("application/json; charset=utf-8");

加了上面的代码后，中文就不会乱码了，效果如下：

{
code: 429,
message: "访问太频繁啦",
route: "yinjihuan"
}

不说了，我还是去提个issues吧： https://github.com/alibaba/Sentinel/issues/733

欢迎加入我的知识星球，一起交流技术，免费学习猿天地的课程（http://cxytiandi.com/course）

PS：目前星球中正在星主的带领下组队学习Spring Cloud，等你哦！