首頁 > 軟體

詳解Spring Cloud Zuul閘道器修改為短連線方法

2022-04-17 13:00:18

一、問題分析

之前在用zuul閘道器的時候,請求幾次然後連線就斷開了。原因是因為http1.1之後,預設走的都是connection=keep-alive 長連線。但沒有心跳維持,顧1分鐘斷開一次。但RestFul一般都是走短連線就行了。因此想著只要修改頭部connection屬性就行了.

就是在過濾器中修改Zuul的RequestContext ctx物件

        //設定請求為短連線
        ctx.addZuulRequestHeader("connection", "close");

再次請求之後發現

​納尼,怎麼還是長連線。那麼說明可能是我的設定沒有生效或者被覆蓋掉了。遇到問題,除了度娘,跟著原始碼走一遭也是最好的方式。花了大半天時間,終於讓我找到原因了。

原因是代理類ProxyRequestHelper裡面有段邏輯。

public boolean isIncludedHeader(String headerName) {
		String name = headerName.toLowerCase();
		RequestContext ctx = RequestContext.getCurrentContext();
		if (ctx.containsKey(IGNORED_HEADERS)) {
			Object object = ctx.get(IGNORED_HEADERS);
			if (object instanceof Collection && ((Collection<?>) object).contains(name)) {
				return false;
			}
		}
		switch (name) {
		case "host":
			if(addHostHeader) {
				return true;
			}
		case "connection":
		case "content-length":
		case "server":
		case "transfer-encoding":
		case "x-application-context":
			return false;
		default:
			return true;
		}
	}

如果頭部存在"connection": "content-length": "server": "transfer-encoding": "x-application-context"這些頭的話都被跳過。不會被設定成功。應該是zuul自己的一個機制把。

二、解決方式

那麼問題找到了,該怎麼修改呢?那麼只能自定義路由了。繼承並重寫這個類。

1、實現自定義路由設定CustomZuulConfig

@Component
public class CustomZuulConfig {

    @Autowired
    ZuulProperties zuulProperties;

    @Value("${servletPath}")
    private String servletPath;

    @Bean
    public CustomRouteLocator routeLocator() {
        CustomRouteLocator routeLocator = new CustomRouteLocator(servletPath, this.zuulProperties);
        return routeLocator;
    }

}

2、實現自定義請求代理工具類 重寫isIncludedHeader方法。

public class CustomRequestHelper extends ProxyRequestHelper {
    @Override
    public boolean isIncludedHeader(String headerName) {
        String name = headerName.toLowerCase();
        RequestContext ctx = RequestContext.getCurrentContext();
        if (ctx.containsKey(IGNORED_HEADERS)) {
            Object object = ctx.get(IGNORED_HEADERS);
            if (object instanceof Collection && ((Collection<?>)object).contains(name)) {
                return false;
            }
        }
        switch (name) {
            case "content-length":
            case "host":
            case "server":
            case "transfer-encoding":
            case "x-application-context":
                return false;
            default:
                return true;
        }
    }
} 

3、重寫實現代理設定類

@Configuration
@EnableZuulProxy
public class CustomZuulProxyConfig extends ZuulProxyAutoConfiguration {

    @Bean
    @Override
    public SimpleHostRoutingFilter simpleHostRoutingFilter(ProxyRequestHelper helper, ZuulProperties zuulProperties,
            ApacheHttpClientConnectionManagerFactory connectionManagerFactory,
            ApacheHttpClientFactory httpClientFactory) {
        CustomRequestHelper customRequestHelper = new CustomRequestHelper();
        return new SimpleHostRoutingFilter(customRequestHelper, zuulProperties, connectionManagerFactory,
                httpClientFactory);
    }

}

然後重新啟動,檢視請求頭。

​修改成功。所以說,學會通過看原始碼查問題,還是很重要的哈~

到此這篇關於詳解Spring Cloud Zuul閘道器修改為短連線方法的文章就介紹到這了,更多相關Spring Cloud Zuul短連線內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com