springcloud gateway自定義斷言規(guī)則,后綴結(jié)尾進(jìn)行路由
因工作需要，需要使用springcloud gateway ,以.html結(jié)尾的進(jìn)行路由進(jìn)行websocket轉(zhuǎn)發(fā)。
gateway自帶的8種路由規(guī)則都不能滿足，故需要自定義斷言規(guī)則。
1.新建一個(gè)路由斷言工廠ExtCheckRoutePredicateFactory
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
@Componentpublic class ExtCheckRoutePredicateFactory extends AbstractRoutePredicateFactory<ExtCheckRoutePredicateFactory.Config> { public ExtCheckRoutePredicateFactory() { super(Config.class); } @Override public Predicate<ServerWebExchange> apply(Config config) { return new Predicate<ServerWebExchange>() { @Override public boolean test(ServerWebExchange serverWebExchange) { String url=serverWebExchange.getRequest().getURI().toString(); if(url.endsWith(".html")){ return true; } return false; } }; } public static class Config{ private String name; public String getName(){ return name; } public void setName(String name){ this.name=name; } }} |
如果以.html結(jié)尾，則匹配此路由
2.修改gateway配置
|
1
2
3
4
5
6
|
gateway: routes: - id: abc uri: http://localhost:8080 predicates: - name: ExtCheck |
ExtCheck即是我們新建斷言工廠的前綴名，自動(dòng)識(shí)別的。
這時(shí)運(yùn)行發(fā)現(xiàn)，系統(tǒng)根本找不到我們自定義的斷言類。
需要第三步
3.修改gateway源碼，將自定義斷言類加到系統(tǒng) predicates里
|
1
2
3
4
5
6
7
8
9
10
|
@Bean public RouteLocator routeDefinitionRouteLocator(GatewayProperties properties, List<GatewayFilterFactory> gatewayFilters, List<RoutePredicateFactory> predicates, RouteDefinitionLocator routeDefinitionLocator, ConfigurationService configurationService) { predicates.add(new ExtCheckRoutePredicateFactory()); return new RouteDefinitionRouteLocator(routeDefinitionLocator, predicates, gatewayFilters, properties, configurationService); } |
再次運(yùn)行，成功根據(jù).html后綴轉(zhuǎn)發(fā)，done!
Gateway自定義路由斷言工廠類
application.yml文件
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
server: port: 7000spring: zipkin: base-url: http://127.0.0.1:9411/ #zipkin server的請求地址 discoveryClientEnabled: false #讓nacos把它當(dāng)成一個(gè)URL，而不要當(dāng)做服務(wù)名 sleuth: sampler: probability: 1.0 #采樣的百分比 application: name: api-gateway cloud: nacos: discovery: server-addr: localhost:8848 # 將gateway注冊到nacos gateway: discovery: locator: enabled: true # 讓gateway從nacos中獲取服務(wù)信息 routes: - id: product_route uri: lb://service-product order: 1 predicates: - Path=/product-serv/** filters: - StripPrefix=1 - id: order_route uri: lb://service-order order: 1 predicates: - Path=/order-serv/** - Age=18,60 filters: - StripPrefix=1 |
路由斷言工廠配置類
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
import lombok.Data;import lombok.NoArgsConstructor;import org.apache.commons.lang3.StringUtils;import org.springframework.cloud.gateway.handler.predicate.AbstractRoutePredicateFactory;import org.springframework.stereotype.Component;import org.springframework.web.server.ServerWebExchange;import java.util.Arrays;import java.util.List;import java.util.function.Predicate;//這是一個(gè)自定義的路由斷言工廠類,要求有兩個(gè)//1 名字必須是 配置+RoutePredicateFactory//2 必須繼承AbstractRoutePredicateFactory<配置類>@Componentpublic class AgeRoutePredicateFactory extends AbstractRoutePredicateFactory<AgeRoutePredicateFactory.Config> { //構(gòu)造函數(shù) public AgeRoutePredicateFactory() { super(Config.class); } //讀取配置文件的中參數(shù)值 給他賦值到配置類中的屬性上 public List<String> shortcutFieldOrder() { //這個(gè)位置的順序必須跟配置文件中的值的順序?qū)?yīng) return Arrays.asList("minAge", "maxAge"); } //斷言邏輯 public Predicate<ServerWebExchange> apply(Config config) { return new Predicate<ServerWebExchange>() { @Override public boolean test(ServerWebExchange serverWebExchange) { //1 接收前臺(tái)傳入的age參數(shù) String ageStr = serverWebExchange.getRequest().getQueryParams().getFirst("age"); //2 先判斷是否為空 if (StringUtils.isNotEmpty(ageStr)) { //3 如果不為空,再進(jìn)行路由邏輯判斷 int age = Integer.parseInt(ageStr); if (age < config.getMaxAge() && age > config.getMinAge()) { return true; } else { return false; } } return false; } }; } //配置類,用于接收配置文件中的對應(yīng)參數(shù) @Data @NoArgsConstructor public static class Config { private int minAge;//18 private int maxAge;//60 }} |
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持服務(wù)器之家。
原文鏈接：https://blog.csdn.net/zhangzhen02/article/details/109082792
