【笔记】Spring Cloud Gateway 实现 gRPC 代理

import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.io.buffer.DataBufferUtils;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
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 GrpcProxyFilter implements GlobalFilter {
 
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        // 如果请求方法不是CONNECT，则继续处理
        if (!exchange.getRequest().getMethod().equals(HttpMethod.CONNECT)) {
            return chain.filter(exchange);
        }
 
        // 构建gRPC代理的目标地址
        URI uri = exchange.getRequest().getURI();
        String host = uri.getHost();
        int port = uri.getPort();
        if (port == -1) {
            if ("http".equals(uri.getScheme())) {
                port = 80;
            } else if ("https".equals(uri.getScheme())) {
                port = 443;
            }
        }
 
        // 修改原始请求的方法为GET，因为gRPC通常使用HTTP/2进行通信
        exchange.getRequest().mutate().method(HttpMethod.GET).build();
 
        // 构建gRPC请求的目标地址
        URI newUri = URI.create(String.format("http://%s:%d", host, port));
        ServerHttpRequest request = exchange.getRequest().mutate().uri(newUri).build();
 
        // 转发请求到gRPC服务器
        ServerHttpResponse response = exchange.getResponse();
        HttpHeaders headers = response.getHeaders();
        // 设置HTTP响应头，以便客户端理解响应是gRPC转发的结果
        headers.set("grpc-status", "0"); // 设置gRPC状态为OK
        headers.set("grpc-encoding", "identity"); // 设置gRPC编码方式为identity
 
        // 转发请求到gRPC服务器并返回响应
        return chain.filter(exchange.mutate().request(request).build());
    }
}

这个代码示例展示了如何在Spring Cloud Gateway中实现一个简单的gRPC代理过滤器。它修改了原始请求的方法，并转发请求到相应的gRPC服务器。同时，它设置了适当的HTTP响应头，以便客户端可以理解并正确处理响应。这个示例提供了一个基本框架，开发者可以根据具体需求进行扩展和定制。