在 Go 中使用 gRPC 的优雅错误处理

grpc 提供了优雅的错误处理机制，允许将自定义错误转换为 grpc 错误并从 grpc 错误中提取自定义错误，以方便在分布式系统中进行错误处理：自定义错误转换为 grpc 错误：使用 codes.new 和 errors.new 函数创建代码和错误。从 grpc 错误提取自定义错误：使用 status.fromerror 和 errors.unwrap 函数获取代码和自定义错误。

在 Go 中使用 gRPC 的优雅错误处理

在分布式系统中，优雅地处理错误非常重要。gRPC 是一个流行的用于构建分布式系统的框架，它提供了强大的错误处理机制。

gRPC 错误处理

gRPC 返回error类型的值以表示错误。gRPC 错误类型实现了github.com/golang/protobuf/ptypes/any接口，允许序列化和反序列化自定义错误。

将自定义错误转换为 gRPC 错误

要将自定义错误转换为 gRPC 错误，请使用codes.New函数创建code.Code并使用errors.New创建error：

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

func ToGRPCError(code codes.Code, err error) error {
    return status.Error(code, err.Error())
}

从 gRPC 错误中提取自定义错误

要从 gRPC 错误中提取自定义错误，请使用errors.Unwrap函数：

import (
    "errors"
)

func FromGRPCError[T any](err error) (error, bool) {
    grpcStatus, ok := status.FromError(err)
    if !ok {
        return nil, false
    }
    if !errors.Is(err, grpcStatus.Proto()) {
        return nil, false
    }
    return errors.Unwrap(grpcStatus.Proto()), true
}

实战案例

示例 gRPC 服务：

package main

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

type GreeterService struct {}

func (s *GreeterService) Greet(ctx context.Context, req *GreetRequest) (*GreetResponse, error) {
    if req.Name == "" {
        return nil, ToGRPCError(codes.InvalidArgument, errors.New("invalid"))
    }
    return &GreetResponse{Message: "Hello, " + req.Name}, nil
}

示例 gRPC 客户端：

package main

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

func main() {
    // ...
    client, err := createClient()
    if err != nil {
        // Handle initialization error
    }
    req := &GreetRequest{Name: ""}
    resp, err := client.Greet(context.Background(), req)
    if err != nil {
        grpcStatus, ok := status.FromError(err)
        if ok && grpcStatus.Code() == codes.InvalidArgument {
            // Handle invalid name error
        } else {
            // Handle other errors
        }
    }
    // ...
}

以上就是在 Go 中使用 gRPC 的优雅错误处理的详细内容，更多请关注其它相关文章！