如何在 Go 中断言 gRPC 错误代码客户端

Question

给定以下 gRPC 服务器端代码：

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

....

func (s *Router) Assign(ctx context.Context, req *api.Request(*api.Response, error) {

    return nil, status.Errorf(codes.PermissionDenied,
}

....

在客户端断言错误是代码 = codes.PermissionDenied 的推荐技术是什么？

Answer 1

假设你的服务器像这样返回codes.PermissionDenined

...
return nil, status.Error(codes.PermissionDenied, "PERMISSION_DENIED_TEXT")

如果您的客户端也是 Golang 也可以使用 status 库函数 FromError 来解析错误。我使用开关来确定返回的错误代码，如下所示

// client
    assignvar, err := s.MyFunctionCall(ctx, ...)
    if err != nil {
        if e, ok := status.FromError(err); ok {
            switch e.Code() {
            case codes.PermissionDenied:
                fmt.Println(e.Message()) // this will print PERMISSION_DENIED_TEST
            case codes.Internal:
                fmt.Println("Has Internal Error")
            case codes.Aborted:
                fmt.Println("gRPC Aborted the call")
            default:
                fmt.Println(e.Code(), e.Message())
            }
        }
        else {
            fmt.Printf("not able to parse error returned %v", err)
        }
    }