处理 rpc 超时错误：使用 context 取消：设置带有超时的上下文，超时后自动取消请求。使用 clientconn 选项：在创建 clientconn 时指定超时时间。使用 unary 拦截器：在 rpc 调用前和后修改请求和响应，并可按需设置超时。

如何在 Go 中处理 RPC 超时错误

概述

在分布式系统中，RPC（远程过程调用）超时是常见的错误场景。Go 提供了多种机制来处理和捕获这些错误。

超时处理方法

上下文取消

使用 Golang 的 context.Context 可以设置带有超时的上下文，它将在超时后取消请求。

立即学习“go语言免费学习笔记（深入）”；

import (
    "context"
    "time"
)

func contextCancelTimeout() {
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()

    // 执行 RPC 调用
    // ...
}

使用 ClientConn 选项

grpc.WithTimeout() 选项可以在创建 ClientConn 时指定超时时间。

import (
    "google.golang.org/grpc"
)

func clientConnTimeout() {
    // 设置超时时间（以毫秒为单位）
    timeout := 5000

    // 使用 WithTimeout() 选项创建 ClientConn
    conn, err := grpc.Dial("localhost:8080", grpc.WithTimeout(timeout*time.Millisecond))
    // ...
}

使用 UnaryInterceptor

gRPC 的 UnaryInterceptor 允许在 RPC 调用之前和之后修改请求和响应。

import (
    "context"
    "grpc"
)

func unaryInterceptorTimeout() {
    timeoutInterceptor := func(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker) error {
        ctx, cancel := context.WithTimeout(ctx, 5*time.Second)
        defer cancel()

        // 执行 RPC 调用
        err := invoker(ctx, method, req, reply, cc)
        // ...

        return err
    }
    // 注册拦截器
    grpc.RegisterUnaryClientInterceptor(timeoutInterceptor)
}

实战案例

场景：从一个远程服务获取用户数据，并限制请求时间为 3 秒。

代码示例：

package main

import (
    "context"
    "log"
    "time"

    pb "github.com/example/grpc/user"
    "google.golang.org/grpc"
)

func main() {
    // 设置超时时间
    timeout := 3 * time.Second
    ctx, cancel := context.WithTimeout(context.Background(), timeout)
    defer cancel()

    // 创建 gRPC 连接
    conn, err := grpc.Dial("localhost:8080", grpc.WithTimeout(timeout))
    if err != nil {
        log.Fatalf("failed to dial: %v", err)
    }
    defer conn.Close()

    // 创建 gRPC 客户端
    userClient := pb.NewUserServiceClient(conn)

    // 执行 RPC 调用
    resp, err := userClient.GetUser(ctx, &pb.GetUserRequest{Id: 1})
    if err != nil {
        if ctx.Err() == context.Canceled {
            log.Println("request timed out")
        } else {
            log.Fatalf("failed to get user: %v", err)
        }
    }

    log.Printf("Got user: %s", resp.GetName())
}