gRPC客户端/Stub跨Goroutine使用及复用可行性问询

问题描述 #

我正在学习Protobuf与gRPC的Go基础教程，了解到ClientConn支持并发安全使用，但未找到关于生成的Stub/客户端的相关说明。现有场景：接收HTTP POST请求后，通过gRPC以Protobuf格式转发数据并响应HTTP请求，请问是每次处理请求时创建客户端/Stub，还是可在HTTP服务器初始化时创建并传入Handler？以下两个代码示例中，第二种是否可行？

示例1 #

var opts []grpc.DialOption
conn, err := grpc.Dial("127.0.0.1:1234", opts...)
if err != nil {
    log.Panic(err)
}
defer conn.Close()
grpcService := service.Service{GRPC: conn}
http.HandleFunc("/", util.ValidateRequest(grpcService.HandleRoot))

// listen to port
http.ListenAndServe(fmt.Sprintf("%s:%d", viper.GetString("server.address"), viper.GetInt32("server.port")), nil)

示例2 #

var opts []grpc.DialOption
conn, err := grpc.Dial("127.0.0.1:1234", opts...)
if err != nil {
    log.Panic(err)
}
defer conn.Close()
client := pb.NewEventShipperClient(conn)
grpcService := service.Service{GRPC: conn, Client: client}
http.HandleFunc("/", util.ValidateRequest(grpcService.HandleRoot))

// listen to port
http.ListenAndServe(fmt.Sprintf("%s:%d", viper.GetString("server.address"), viper.GetInt32("server.port")), nil)

解答 #

• gRPC生成的Stub（比如示例中的pb.NewEventShipperClient创建的客户端）是并发安全的，完全可以在HTTP服务初始化阶段创建一次，之后在所有请求Handler中复用，不需要每次处理请求时重新创建。
• 示例2完全可行，而且是更优的实践方案。Stub本质上是对ClientConn的轻量封装，创建过程几乎没有额外资源开销，但提前初始化并复用Stub可以避免重复创建的冗余操作，提升服务性能。
• 原理说明：Stub内部仅持有ClientConn的引用，所有RPC调用最终都会通过ClientConn执行，而ClientConn本身已经被官方明确为并发安全，因此复用Stub不存在线程安全问题。
• 对比示例1：如果示例1的grpcService.HandleRoot方法内部每次都会创建新的Stub，虽然功能上能正常工作，但属于不必要的重复操作，效率不如示例2的复用方式。
• 注意事项：确保ClientConn在整个HTTP服务生命周期内保持有效即可，你代码中defer conn.Close()的位置是合理的——因为ListenAndServe是阻塞调用，只有当服务退出时才会执行defer语句关闭连接，不会影响正常请求处理。

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