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
}
}
// ...
}