Go语言gRPC客户端检测服务器断开连接方案咨询

首先明确你的核心需求：基于现有的SayHello一元RPC服务，实现客户端与服务端的长连接维持，服务端宕机后自动重连并重新调用SayHello。

不推荐的方案分析 #

• 方案2（新增ClientStreaming API）：完全没必要为了保活和重连新增API，会额外增加服务端和客户端的复杂度，违背了尽量复用现有服务定义的原则。

推荐的方案组合 #

1. 利用gRPC内置的连接重试机制 #

gRPC本身已经内置了连接层的重试逻辑，不需要自己通过statsHandler来实现。你只需要在grpc.Dial时配置合适的拨号参数：

conn, err := grpc.Dial(
    "server-address:port",
    grpc.WithInsecure(), // 生产环境请用TLS相关配置
    grpc.WithConnectParams(grpc.ConnectParams{
        Backoff: grpc.BackoffConfig{
            BaseDelay:  1 * time.Second,   // 初始重试间隔
            Multiplier: 1.5,               // 间隔倍增系数
            Jitter:     0.2,               // 随机抖动，避免惊群效应
            MaxDelay:   10 * time.Second,  // 最大重试间隔
        },
        MinConnectTimeout: 2 * time.Second,
    }),
)

这个配置会让gRPC客户端在连接断开后自动按照退避策略尝试重连，不需要额外的statsHandler处理连接事件。

2. 针对SayHello请求的重试封装 #

连接重连只是底层层面的恢复，你还需要对SayHello的调用做上层重试封装，处理连接断开时的RPC错误：

import (
    "context"
    "fmt"
    "math"
    "time"

    "google.golang.org/grpc/codes"
    "google.golang.org/grpc/status"
    pb "your/path/to/proto" // 替换为你的proto包路径
)

func retrySayHello(client pb.GreeterClient, req *pb.HelloRequest, maxRetries int) (*pb.HelloReply, error) {
    var err error
    var resp *pb.HelloReply
    for i := 0; i < maxRetries; i++ {
        resp, err = client.SayHello(context.Background(), req)
        if err == nil {
            return resp, nil
        }
        // 判断是否是可重试的错误（比如连接错误、服务不可用）
        if status.Code(err) == codes.Unavailable || status.Code(err) == codes.DeadlineExceeded {
            // 指数退避重试
            backoff := time.Duration(math.Pow(2, float64(i))) * time.Second
            time.Sleep(backoff)
            continue
        }
        // 非可重试错误直接返回
        return nil, err
    }
    return nil, fmt.Errorf("max retries exceeded: %v", err)
}

3. 可选：连接保活机制 #

如果需要维持长连接避免被中间件（如负载均衡、防火墙）断开，可以添加保活配置：

conn, err := grpc.Dial(
    "server-address:port",
    grpc.WithInsecure(),
    grpc.WithKeepaliveParams(grpc.KeepaliveParams{
        Time:                30 * time.Second, // 每隔30秒发送一次保活ping
        Timeout:             5 * time.Second,  // ping超时时间
        PermitWithoutStream: true,             // 即使没有活跃RPC也发送ping
    }),
)

总结 #

• 优先使用gRPC内置的连接重试和保活机制，不需要自己实现statsHandler的连接重试逻辑，避免重复造轮子。
• 对SayHello请求做上层的重试封装，处理RPC调用时的可重试错误。
• 不要新增ClientStreaming API，完全是冗余设计。

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