How to assert gRPC error codes client side in Go

在Go中处理gRPC错误代码，主要涉及到两个库：google.golang.org/grpc/status 和 google.golang.org/grpc/codes。断言gRPC错误代码通常是为了根据不同的错误类型做出相应的处理。下面是一个详细的步骤说明和示例：

步骤 1: 捕获错误

当你从一个gRPC调用中接收到错误时，首先需要检查这个错误是否为nil。如果不是nil，则进一步处理该错误。

go
resp, err := client.SomeRPCMethod(ctx, req)
if err != nil {
    // 处理错误
}

步骤 2: 断言错误类型

使用status包中的FromError函数将错误类型转换为*status.Status，这样可以让我们获取错误的具体信息，如错误代码和错误消息。

go
st, ok := status.FromError(err)
if !ok {
    // err 不是一个 *status.Status 类型的错误, 可能是其他类型的错误
    return fmt.Errorf("unexpected error type: %v", err)
}

步骤 3: 获取并断言错误代码

通过st.Code()方法获取错误代码，并通过判断语句来匹配不同的错误代码。这里的codes包包含了所有的gRPC标准错误代码。

go
switch st.Code() {
case codes.NotFound:
    // 处理未找到的情况
    fmt.Println("资源未找到:", st.Message())
case codes.PermissionDenied:
    // 处理权限不足的情况
    fmt.Println("权限不足:", st.Message())
default:
    // 处理其他类型的错误
    fmt.Println("其他错误:", st.Message())
}

示例

组合上述步骤，以下是一个完整的示例，这个示例模拟了一个客户端调用gRPC服务的过程，并根据返回的错误类型做出相应处理。

go
package main

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

func main() {
    conn, err := grpc.Dial("your-grpc-server-address", grpc.WithInsecure())
    if err != nil {
        panic(err)
    }
    defer conn.Close()

    client := NewYourClientServiceClient(conn)

    req := &YourRequest{}  // 假设YouRequest是gRPC请求的类型
    ctx := context.Background()
    resp, err := client.YourMethod(ctx, req)
    if err != nil {
        st, ok := status.FromError(err)
        if !ok {
            fmt.Printf("非gRPC错误: %v\n", err)
            return
        }
        switch st.Code() {
        case codes.NotFound:
            fmt.Println("资源未找到:", st.Message())
        case codes.PermissionDenied:
            fmt.Println("权限不足:", st.Message())
        default:
            fmt.Println("其他错误:", st.Message())
        }
        return
    }

    fmt.Println("响应:", resp)
}

在这个示例中，我们首先尝试与gRPC服务建立连接，然后发送一个请求。如果遇到错误，我们将错误断言为*status.Status类型，进而获取错误代码并对不同类型的错误进行相应的处理。这样的处理方式可以让错误管理更加清晰和有条理。