如何在Micronaut中实现通用gRPC请求转发至遗留服务？

要实现任意gRPC请求的通用转发，不能依赖特定服务的Stub，得用gRPC底层的ManagedChannel和通用调用API来处理所有服务方法。核心思路是拦截入站请求后，通过通道直接构造对应的客户端调用，将请求、元数据和响应双向转发。

完整实现示例 #

import io.grpc.*;
import jakarta.inject.Inject;
import jakarta.inject.Named;
import jakarta.inject.Singleton;
import java.util.concurrent.Executor;

@Singleton
public class LegacyGrpcProxyInterceptor implements ServerInterceptor {

    // 注入通用的gRPC通道，而非特定服务的Stub
    private final ManagedChannel legacyChannel;
    private final Executor executor;

    @Inject
    public LegacyGrpcProxyInterceptor(@Named("legacy-grpc-channel") ManagedChannel legacyChannel,
                                      @Named("grpc-proxy-executor") Executor executor) {
        this.legacyChannel = legacyChannel;
        this.executor = executor;
    }

    @Override
    public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(ServerCall<ReqT, RespT> serverCall,
                                                                 Metadata headers,
                                                                 ServerCallHandler<ReqT, RespT> next) {
        // 获取当前请求的方法描述符
        MethodDescriptor<ReqT, RespT> methodDescriptor = serverCall.getMethodDescriptor();

        // 创建指向遗留服务的客户端调用
        ClientCall<ReqT, RespT> clientCall = legacyChannel.newCall(methodDescriptor, CallOptions.DEFAULT);

        // 自定义ServerCall，将服务端的响应转发给客户端
        ServerCall<ReqT, RespT> proxyServerCall = new ForwardingServerCall.SimpleForwardingServerCall<ReqT, RespT>(serverCall) {
            @Override
            public void sendHeaders(Metadata responseHeaders) {
                // 这里可以修改或直接转发遗留服务返回的响应头
                super.sendHeaders(responseHeaders);
            }

            @Override
            public void sendMessage(RespT message) {
                // 将遗留服务的响应消息转发给当前请求的客户端
                super.sendMessage(message);
            }
        };

        // 自定义客户端调用的Listener，将入站请求转发给遗留服务
        ClientCall.Listener<RespT> clientListener = new ClientCall.Listener<RespT>() {
            @Override
            public void onHeaders(Metadata responseHeaders) {
                // 收到遗留服务的响应头后，发送给当前服务的客户端
                proxyServerCall.sendHeaders(responseHeaders);
            }

            @Override
            public void onMessage(RespT message) {
                // 收到遗留服务的响应消息后，发送给当前服务的客户端
                proxyServerCall.sendMessage(message);
            }

            @Override
            public void onClose(Status status, Metadata trailers) {
                // 遗留服务调用结束后，关闭当前服务的调用
                proxyServerCall.close(status, trailers);
            }
        };

        // 启动客户端调用，传入请求头
        clientCall.start(clientListener, headers);

        // 返回自定义的ServerCall.Listener，处理入站请求的转发
        return new ForwardingServerCallListener.SimpleForwardingServerCallListener<ReqT>(next.startCall(proxyServerCall, headers)) {
            @Override
            public void onMessage(ReqT message) {
                // 将客户端发来的请求消息转发给遗留服务
                clientCall.sendMessage(message);
            }

            @Override
            public void onHalfClose() {
                // 客户端发送完请求后，通知遗留服务
                clientCall.halfClose();
            }

            @Override
            public void onCancel() {
                // 当前请求被取消时，取消遗留服务的调用
                clientCall.cancel("Request cancelled by client", null);
                super.onCancel();
            }
        };
    }
}

关键逻辑说明 #

1. 依赖注入调整：放弃特定服务的Stub，改用ManagedChannel——它是gRPC客户端的核心通道，能支持任意服务方法的调用。
2. 方法描述符复用：通过serverCall.getMethodDescriptor()获取当前请求的方法元信息，直接复用它构造客户端调用，保证请求方法和遗留服务完全匹配。
3. 双向转发处理：
   • 自定义ServerCall：负责将遗留服务的响应头、响应消息回传给当前请求的客户端。
   • 自定义ClientCall.Listener：监听遗留服务的响应，同步到当前服务的调用上下文。
   • 自定义ServerCall.Listener：将客户端发来的请求消息、状态（半关闭、取消）转发给遗留服务。
4. 通用兼容所有调用类型：不管是一元RPC、服务器流式、客户端流式还是双向流式，这套逻辑都能自动适配，因为所有类型的gRPC调用都基于ClientCall和ServerCall的统一接口。

注意事项 #

• 元数据处理：如果需要修改请求头/响应头（比如添加转发标识），可以在onHeaders或sendHeaders方法中对Metadata进行修改。
• 错误处理：要确保遗留服务的调用状态（比如异常、取消）能正确同步到当前服务的客户端，避免请求挂起。
• 资源管理：ManagedChannel需要在应用启动时初始化、关闭时销毁，避免资源泄漏。

内容的提问来源于stack exchange，提问作者frblazquez