轻量级的Spring Cloud Gateway实践，实现api和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.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;
 
@Component
public class WebsocketAndApiForwardingFilter implements GlobalFilter, Ordered {
 
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        ServerHttpRequest request = exchange.getRequest();
 
        // 检查是否为Websocket请求
        if (isWebsocketRequest(request)) {
            // 这里可以添加处理Websocket请求的逻辑
            return handleWebsocketRequest(exchange);
        }
 
        // 继续处理API请求
        return chain.filter(exchange);
    }
 
    private boolean isWebsocketRequest(ServerHttpRequest request) {
        // 检查请求头中的Upgrade字段，查看是否为Websocket请求
        return "websocket".equalsIgnoreCase(request.getHeaders().getUpgrade());
    }
 
    private Mono<Void> handleWebsocketRequest(ServerWebExchange exchange) {
        // 这里实现转发Websocket请求的逻辑
        // 例如，可以直接返回一个特定的响应或者调用特定的服务处理
        return Mono.empty(); // 暂时返回一个空的Mono，表示不做任何处理
    }
 
    @Override
    public int getOrder() {
        // 设置过滤器的顺序，数字越小，优先级越高
        return -1;
    }
}

这个代码示例展示了如何检查是否为Websocket请求，并根据请求类型来转发或处理请求。这个过滤器可以被添加到Spring Cloud Gateway中，用于实现基于请求类型的路由分发。