springcloud gateway自定義斷言規則,后綴結尾進行路由
因工作需要,需要使用springcloud gateway ,以.html結尾的進行路由進行websocket轉發。
gateway自帶的8種路由規則都不能滿足,故需要自定義斷言規則。
1.新建一個路由斷言工廠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
|
@Component public 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結尾,則匹配此路由
2.修改gateway配置
1
2
3
4
5
6
|
gateway: routes: - id: abc uri: http: //localhost:8080 predicates: - name: ExtCheck |
ExtCheck即是我們新建斷言工廠的前綴名,自動識別的。
這時運行發現,系統根本找不到我們自定義的斷言類。
需要第三步
3.修改gateway源碼,將自定義斷言類加到系統 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); } |
再次運行,成功根據.html后綴轉發,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: 7000 spring: zipkin: base-url: http: //127.0.0.1:9411/ #zipkin server的請求地址 discoveryClientEnabled: false #讓nacos把它當成一個URL,而不要當做服務名 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中獲取服務信息 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; //這是一個自定義的路由斷言工廠類,要求有兩個 //1 名字必須是 配置+RoutePredicateFactory //2 必須繼承AbstractRoutePredicateFactory<配置類> @Component public class AgeRoutePredicateFactory extends AbstractRoutePredicateFactory<AgeRoutePredicateFactory.Config> { //構造函數 public AgeRoutePredicateFactory() { super (Config. class ); } //讀取配置文件的中參數值 給他賦值到配置類中的屬性上 public List<String> shortcutFieldOrder() { //這個位置的順序必須跟配置文件中的值的順序對應 return Arrays.asList( "minAge" , "maxAge" ); } //斷言邏輯 public Predicate<ServerWebExchange> apply(Config config) { return new Predicate<ServerWebExchange>() { @Override public boolean test(ServerWebExchange serverWebExchange) { //1 接收前臺傳入的age參數 String ageStr = serverWebExchange.getRequest().getQueryParams().getFirst( "age" ); //2 先判斷是否為空 if (StringUtils.isNotEmpty(ageStr)) { //3 如果不為空,再進行路由邏輯判斷 int age = Integer.parseInt(ageStr); if (age < config.getMaxAge() && age > config.getMinAge()) { return true ; } else { return false ; } } return false ; } }; } //配置類,用于接收配置文件中的對應參數 @Data @NoArgsConstructor public static class Config { private int minAge; //18 private int maxAge; //60 } } |
以上為個人經驗,希望能給大家一個參考,也希望大家多多支持服務器之家。
原文鏈接:https://blog.csdn.net/zhangzhen02/article/details/109082792