添加websocket转发

gateway/src/main/java/com/huaxu/gateway/config/WebsocketHandler.java

@@ -0,0 +1,53 @@
+package com.huaxu.gateway.config;
+
+import org.springframework.cloud.gateway.filter.GatewayFilterChain;
+import org.springframework.cloud.gateway.filter.GlobalFilter;
+import org.springframework.core.Ordered;
+import org.springframework.stereotype.Component;
+import org.springframework.web.server.ServerWebExchange;
+import org.springframework.web.util.UriComponentsBuilder;
+import reactor.core.publisher.Mono;
+
+import java.net.URI;
+
+import static org.springframework.cloud.gateway.support.ServerWebExchangeUtils.GATEWAY_REQUEST_URL_ATTR;
+@Component
+public class WebsocketHandler implements GlobalFilter, Ordered{
+    private final static String DEFAULT_FILTER_PATH = "/uims-message/webSocket";
+
+    /**
+     *
+     * @param exchange ServerWebExchange是一个HTTP请求-响应交互的契约。提供对HTTP请求和响应的访问，
+     *                 并公开额外的 服务器 端处理相关属性和特性，如请求属性
+     * @param chain
+     * @return
+     */
+    @Override
+    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
+        String upgrade = exchange.getRequest().getHeaders().getUpgrade();
+
+        URI requestUrl = exchange.getRequiredAttribute(GATEWAY_REQUEST_URL_ATTR);
+
+        String scheme = requestUrl.getScheme();
+
+        if (!"ws".equals(scheme) && !"wss".equals(scheme)) {
+            return chain.filter(exchange);
+        } else if (DEFAULT_FILTER_PATH.contains(requestUrl.getPath())) {
+            String wsScheme = convertWsToHttp(scheme);
+            URI wsRequestUrl = UriComponentsBuilder.fromUri(requestUrl).scheme(wsScheme).build().toUri();
+            exchange.getAttributes().put(GATEWAY_REQUEST_URL_ATTR, wsRequestUrl);
+        }
+        return chain.filter(exchange);
+    }
+
+    @Override
+    public int getOrder() {
+        return Ordered.LOWEST_PRECEDENCE - 2;
+    }
+
+    static String convertWsToHttp(String scheme) {
+        scheme = scheme.toLowerCase();
+        return "ws".equals(scheme) ? "http" : "wss".equals(scheme) ? "https" : scheme;
+    }
+
+}

gateway/src/main/resources/application-dev.properties

@@ -27,5 +27,16 @@ spring.cloud.gateway.routes[2].uri=lb://sms-water
 spring.cloud.gateway.routes[2].predicates[0]=Path=/sms-water/**
 spring.cloud.gateway.routes[2].filters[0]=StripPrefix=1
 
+spring.cloud.gateway.routes[3].id=uims-message
+spring.cloud.gateway.routes[3].uri=lb://uims-message
+spring.cloud.gateway.routes[3].predicates[0]=Path=/uims-message/**
+spring.cloud.gateway.routes[3].filters[0]=StripPrefix=1
+
+spring.cloud.gateway.routes[4].id=uims-message-socket
+spring.cloud.gateway.routes[4].uri=lb:ws://uims-message
+spring.cloud.gateway.routes[4].predicates[0]=Path=/uims-message/webSocket/**
+spring.cloud.gateway.routes[4].filters[0]=StripPrefix=1
+
+
 spring.servlet.multipart.max-file-size=100MB
 spring.servlet.multipart.max-request-size=100MB