first commit

pkg/error/interceptor.go

@@ -0,0 +1,52 @@
+package error
+
+import (
+	"context"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+// WrapErr 返回gRPC状态码包装后的业务错误
+func WrapErr(err error) error {
+	if err == nil {
+		return nil
+	}
+
+	switch e := err.(type) {
+	case interface{ GRPCStatus() *status.Status }:
+		return e.GRPCStatus().Err()
+	case *Error:
+		return status.Error(codes.Code(-e.Code()), e.Error())
+	default:
+		return status.Error(codes.Unknown, err.Error())
+	}
+}
+
+// ErrInterceptor 业务错误服务端一元拦截器
+func ErrInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
+	resp, err := handler(ctx, req)
+	return resp, WrapErr(err)
+}
+
+// ErrClientInterceptor 业务错误客户端一元拦截器
+func ErrClientInterceptor(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
+	err := invoker(ctx, method, req, reply, cc, opts...)
+	return UnwrapErr(err)
+}
+
+// UnwrapErr 返回gRPC状态码解包后的业务错误
+func UnwrapErr(err error) error {
+	if err == nil {
+		return nil
+	}
+
+	s, _ := status.FromError(err)
+	c := int(s.Code())
+
+	if _, ok := errorMsg[-c]; ok {
+		return NewError(-c)
+	}
+
+	return err
+}