如何在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)
        }
    }

Answer 2

您还可以使用status.Code, 直接检查错误代码，而无需Status先获取值。例如：

if err != nil {
    switch status.Code(err) {
    case codes.NotFound:
        // handle not found

    case codes.FailedPrecondition:
        // handle failed precondition

    default:
        // handle other status codes
    }
}