spring cloud gateway网关转发websocket请求

Spring Cloud Gateway 默认不支持 WebSocket，因为 WebSocket 是一个持久化的连接，而 Spring Cloud Gateway 当前版本（2021 年初）不支持 WebSocket 路由。

但是，你可以使用 Spring Cloud Gateway 的过滤器（Filter）来尝试实现 WebSocket 的转发。一个可能的解决方案是自定义一个过滤器来处理 WebSocket 的握手请求，并将其转发到后端服务。

以下是一个简化的例子，展示了如何创建一个自定义的网关过滤器来处理 WebSocket 的转发：

import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;
 
import java.net.URI;
 
public class WebsocketGlobalFilter implements GlobalFilter, Ordered {
 
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        ServerHttpRequest request = exchange.getRequest();
 
        // 检查是否为 WebSocket 握手请求
        if ("websocket".equals(request.getHeaders().getUpgrade())) {
            // 修改原始请求的 URL 以指向后端服务地址
            URI url = request.getURI();
            String newUrl = "ws://your-websocket-service-url"; // 替换为你的WebSocket服务地址
            ServerHttpRequest newRequest = request.mutate().uri(URI.create(newUrl)).build();
 
            // 使用新的 URL 进行转发
            return chain.filter(exchange.mutate().request(newRequest).build());
        }
 
        // 不是 WebSocket 握手请求则继续正常的 Gateway 处理流程
        return chain.filter(exchange);
    }
 
    @Override
    public int getOrder() {
        // 确保这个过滤器在 WebSocketHandler 之前执行
        return -1;
    }
}

然后，你需要将这个过滤器注册到你的 Spring Cloud Gateway 路由中：

import org.springframework.cloud.gateway.route.RouteLocator;
import org.springframework.cloud.gateway.builder.Routes;
 
@Configuration
public class GatewayConfig {
 
    @Bean
    public RouteLocator customRouteLocator(RouteLocatorBuilder builder) {
        return builder.routes()
                .route("websocket_route", r -> r.path("/ws/**")
                        .filters(f -> f.filter(new WebsocketGlobalFilter()))
                        .uri("your-websocket-service-uri")) // 替换为你的WebSocket服务地址
                .build();
    }
}

请注意，这个示例代码是基于假设的条件编写的，并且可能需要进行一些额外的配置或者错误处理。此外，WebSocket 的转发可能会遇到与 WebSocket 协议相关的复杂问题，如协议不匹配、握手失败等。在实际应用中，你可能需要对 WebSocket 的握手过程和消息传递做更深入的了解。