gRPC双向流TCP断开异常处理及客户端重连方案咨询

核心结论 #

gRPC 自身不会自动恢复中断的流连接，必须手动实现重连逻辑。当TCP连接因服务器崩溃重启等原因断开时，已建立的流会进入错误状态，无法自动恢复，需要客户端重新发起流请求。

重连实现方案 #

基于现有代码，通过嵌套循环+错误判断+重试间隔的方式实现重连，具体修改如下：

import (
    "context"
    "fmt"
    "time"
    "google.golang.org/grpc/codes"
    "google.golang.org/grpc/status"
)

func handler() {
    req := rpc.LiveReq{}
    // 外层循环：负责触发重连逻辑
    for {
        // 每次重连创建独立上下文，避免旧上下文干扰
        ctx, cancel := context.WithCancel(context.Background())
        defer cancel()

        // 重新发起流请求
        stream, err := slave.GetLiveClient().Dial(ctx, &req)
        if err != nil {
            fmt.Printf("流连接建立失败: %v\n", err)
            // 等待后重试，可根据需求调整间隔
            time.Sleep(3 * time.Second)
            continue
        }

        // 内层循环：正常接收流数据
        for {
            resp, err := stream.Recv()
            if err != nil {
                // 解析gRPC错误码，判断是否需要重连
                s, ok := status.FromError(err)
                if ok {
                    switch s.Code() {
                    case codes.Canceled, codes.Unavailable, codes.ResourceExhausted:
                        fmt.Printf("流连接中断，准备重连: %v\n", err)
                    default:
                        fmt.Printf("数据接收错误: %v\n", err)
                    }
                } else {
                    fmt.Printf("接收异常: %v\n", err)
                }
                // 退出内层循环，触发外层重连
                break
            }
            fmt.Printf("收到数据: %+v\n", resp)
        }

        // 释放当前流的上下文资源
        cancel()
        // 重试前等待，避免频繁请求
        time.Sleep(3 * time.Second)
    }
}

关键实现细节 #

1. 嵌套循环结构：外层循环处理重连逻辑，每次流断开后重新发起Dial请求创建新流；内层循环负责正常接收流数据。
2. 错误类型判断：通过status.FromError解析gRPC错误码，针对连接类错误（如Unavailable、ResourceExhausted）触发重连，其他错误可根据业务需求处理。
3. 重试间隔控制：添加固定等待时间（或实现指数退避策略），避免短时间内频繁重试给服务器造成压力。
4. 上下文管理：每次重连创建独立上下文，流断开时及时调用cancel释放资源，避免内存泄漏。

补充说明 #

gRPC客户端维护的连接池会自动处理无效连接的清理，重新调用Dial时，客户端会尝试创建新的有效连接，无需手动刷新连接池。如果业务需要保留请求状态，需确保重连时能正确传递相关参数。

内容的提问来源于stack exchange，提问作者王伯荣