Go gRPC服务器复用已建立客户端连接实现RPC代理方案问询

问题根源 #

GreeterClient 和 GreeterServer 接口的方法签名不兼容：客户端方法多了可变参数 ...grpc.CallOption，因此无法直接将客户端实例作为服务器实现注册。

方案一：极简包装器（低冗余） #

通过通用包装函数消除参数差异，每个RPC方法只需一行代码完成转发，避免重复编写逻辑。

1. 定义通用Unary方法包装器

import "google.golang.org/grpc"

// unaryProxy 适配客户端与服务器的方法签名差异
func unaryProxy[Req, Resp any](clientFunc func(context.Context, Req, ...grpc.CallOption) (Resp, error)) func(context.Context, Req) (Resp, error) {
    return func(ctx context.Context, req Req) (Resp, error) {
        // 可根据需求添加默认CallOption，比如超时、拦截器
        return clientFunc(ctx, req)
    }
}

2. 实现服务器接口的代理结构体

type GreeterProxy struct {
    Client pb.GreeterClient
}

// 代理SayHello方法
func (p *GreeterProxy) SayHello(ctx context.Context, req *pb.HelloRequest) (*pb.HelloReply, error) {
    return unaryProxy(p.Client.SayHello)(ctx, req)
}

// 若有其他方法，如SayHelloAgain，同样一行完成包装
func (p *GreeterProxy) SayHelloAgain(ctx context.Context, req *pb.HelloRequest) (*pb.HelloReply, error) {
    return unaryProxy(p.Client.SayHelloAgain)(ctx, req)
}

3. 注册代理实例到服务器

func server(port int, client pb.GreeterClient) {
    lis, err := net.Listen("tcp", fmt.Sprintf(":%d", port))
    if err != nil {
        panic(err)
    }

    s := grpc.NewServer()
    // 注册包装后的代理实例
    pb.RegisterGreeterServer(s, &GreeterProxy{Client: client})
    
    if err := s.Serve(lis); err != nil {
        panic(err)
    }
}

方案二：代码生成（完全自动化） #

如果服务方法极多且频繁变动，可通过protoc自定义插件或go generate+模板自动生成代理代码，彻底避免手动维护。

示例模板（基于text/template）：

// {{.ServiceName}}Proxy 代理{{.ServiceName}}客户端到服务器
type {{.ServiceName}}Proxy struct {
    Client pb.{{.ServiceName}}Client
}

{{range .Methods}}
// {{.Name}} 代理{{.Name}}方法
func (p *{{$.ServiceName}}Proxy) {{.Name}}(ctx context.Context, req *pb.{{.RequestType}}) (*pb.{{.ReplyType}}, error) {
    return unaryProxy(p.Client.{{.Name}})(ctx, req)
}
{{end}}

每次proto文件更新后，执行go generate即可自动生成最新的代理代码。

流式方法处理注意 #

对于ServerStream/ClientStream/BidiStream类型的方法，需要单独编写流转发逻辑，因为流式方法的签名差异更大，需要手动处理消息的读写转发。

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