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