mirror of
https://github.com/linka-cloud/grpc.git
synced 2024-11-13 06:26:25 +00:00
Adphi
3fb566cb80
example: add auth metadata interceptor: do not overide metadatas Signed-off-by: Adphi <philippe.adrien.nousse@gmail.com>
288 lines
8.1 KiB
Go
288 lines
8.1 KiB
Go
package errors
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"strings"
|
|
|
|
status2 "google.golang.org/genproto/googleapis/rpc/status"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
"google.golang.org/protobuf/proto"
|
|
"google.golang.org/protobuf/types/known/anypb"
|
|
)
|
|
|
|
func Canceled(err error) error {
|
|
return status.Error(codes.Canceled, err.Error())
|
|
}
|
|
|
|
func InvalidArgument(err error) error {
|
|
return status.Error(codes.InvalidArgument, err.Error())
|
|
}
|
|
func DeadlineExceeded(err error) error {
|
|
return status.Error(codes.DeadlineExceeded, err.Error())
|
|
}
|
|
func NotFound(err error) error {
|
|
return status.Error(codes.NotFound, err.Error())
|
|
}
|
|
func AlreadyExists(err error) error {
|
|
return status.Error(codes.AlreadyExists, err.Error())
|
|
}
|
|
func PermissionDenied(err error) error {
|
|
return status.Error(codes.PermissionDenied, err.Error())
|
|
}
|
|
func ResourceExhausted(err error) error {
|
|
return status.Error(codes.ResourceExhausted, err.Error())
|
|
}
|
|
func FailedPrecondition(err error) error {
|
|
return status.Error(codes.FailedPrecondition, err.Error())
|
|
}
|
|
func Aborted(err error) error {
|
|
return status.Error(codes.Aborted, err.Error())
|
|
}
|
|
func OutOfRange(err error) error {
|
|
return status.Error(codes.OutOfRange, err.Error())
|
|
}
|
|
func Unimplemented(err error) error {
|
|
return status.Error(codes.Unimplemented, err.Error())
|
|
}
|
|
func Internal(err error) error {
|
|
return status.Error(codes.Internal, err.Error())
|
|
}
|
|
func Unavailable(err error) error {
|
|
return status.Error(codes.Unavailable, err.Error())
|
|
}
|
|
func DataLoss(err error) error {
|
|
return status.Error(codes.DataLoss, err.Error())
|
|
}
|
|
func Unauthenticated(err error) error {
|
|
return status.Error(codes.Unauthenticated, err.Error())
|
|
}
|
|
func statusErr(code codes.Code, err error, details ...proto.Message) error {
|
|
return status.FromProto(&status2.Status{Code: int32(code), Message: err.Error(), Details: makeDetails(details...)}).Err()
|
|
}
|
|
func makeDetails(m ...proto.Message) []*anypb.Any {
|
|
var out []*anypb.Any
|
|
for _, v := range m {
|
|
if a, err := anypb.New(v); err == nil {
|
|
out = append(out, a)
|
|
}
|
|
}
|
|
return out
|
|
}
|
|
|
|
func Canceledf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.Canceled, msg, args...)
|
|
}
|
|
func Canceledd(err error, details ...proto.Message) error {
|
|
return statusErr(codes.Canceled, err, details...)
|
|
}
|
|
func InvalidArgumentf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.InvalidArgument, msg, args...)
|
|
}
|
|
func InvalidArgumentd(err error, details ...proto.Message) error {
|
|
return statusErr(codes.InvalidArgument, err, details...)
|
|
}
|
|
func DeadlineExceededf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.DeadlineExceeded, msg, args...)
|
|
}
|
|
func DeadlineExceededd(err error, details ...proto.Message) error {
|
|
return statusErr(codes.DeadlineExceeded, err, details...)
|
|
}
|
|
func NotFoundf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.NotFound, msg, args...)
|
|
}
|
|
func NotFoundd(err error, details ...proto.Message) error {
|
|
return statusErr(codes.NotFound, err, details...)
|
|
}
|
|
func AlreadyExistsf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.AlreadyExists, msg, args...)
|
|
}
|
|
func AlreadyExistsd(err error, details ...proto.Message) error {
|
|
return statusErr(codes.AlreadyExists, err, details...)
|
|
}
|
|
func PermissionDeniedf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.PermissionDenied, msg, args...)
|
|
}
|
|
func PermissionDeniedd(err error, details ...proto.Message) error {
|
|
return statusErr(codes.PermissionDenied, err, details...)
|
|
}
|
|
func ResourceExhaustedf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.ResourceExhausted, msg, args...)
|
|
}
|
|
func ResourceExhaustedd(err error, details ...proto.Message) error {
|
|
return statusErr(codes.ResourceExhausted, err, details...)
|
|
}
|
|
func FailedPreconditionf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.FailedPrecondition, msg, args...)
|
|
}
|
|
func FailedPreconditiond(err error, details ...proto.Message) error {
|
|
return statusErr(codes.FailedPrecondition, err, details...)
|
|
}
|
|
func Abortedf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.Aborted, msg, args...)
|
|
}
|
|
func Abortedd(err error, details ...proto.Message) error {
|
|
return statusErr(codes.Aborted, err, details...)
|
|
}
|
|
func OutOfRangef(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.OutOfRange, msg, args...)
|
|
}
|
|
func OutOfRanged(err error, details ...proto.Message) error {
|
|
return statusErr(codes.OutOfRange, err, details...)
|
|
}
|
|
func Unimplementedf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.Unimplemented, msg, args...)
|
|
}
|
|
func Unimplementedd(err error, details ...proto.Message) error {
|
|
return statusErr(codes.Unimplemented, err, details...)
|
|
}
|
|
func Internalf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.Internal, msg, args...)
|
|
}
|
|
func Internald(err error, details ...proto.Message) error {
|
|
return statusErr(codes.Internal, err, details...)
|
|
}
|
|
func Unavailablef(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.Unavailable, msg, args...)
|
|
}
|
|
func Unavailabled(err error, details ...proto.Message) error {
|
|
return statusErr(codes.Unavailable, err, details...)
|
|
}
|
|
func DataLossf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.DataLoss, msg, args...)
|
|
}
|
|
func DataLossd(err error, details ...proto.Message) error {
|
|
return statusErr(codes.DataLoss, err, details...)
|
|
}
|
|
func Unauthenticatedf(msg string, args ...interface{}) error {
|
|
return status.Errorf(codes.Unauthenticated, msg, args...)
|
|
}
|
|
func Unauthenticatedd(err error, details ...proto.Message) error {
|
|
return statusErr(codes.Unauthenticated, err, details...)
|
|
}
|
|
|
|
func IsCanceled(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.Canceled || IsCanceled(err)
|
|
}
|
|
func IsUnknown(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.Unknown
|
|
}
|
|
func IsInvalidArgument(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.InvalidArgument
|
|
}
|
|
func IsDeadlineExceeded(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.DeadlineExceeded || IsContextDeadlineExceeded(err)
|
|
}
|
|
func IsNotFound(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.NotFound
|
|
}
|
|
func IsAlreadyExists(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.AlreadyExists
|
|
}
|
|
func IsPermissionDenied(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.PermissionDenied
|
|
}
|
|
func IsResourceExhausted(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.ResourceExhausted
|
|
}
|
|
func IsFailedPrecondition(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.FailedPrecondition
|
|
}
|
|
func IsAborted(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.Aborted
|
|
}
|
|
func IsOutOfRange(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.OutOfRange
|
|
}
|
|
func IsUnimplemented(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.Unimplemented
|
|
}
|
|
func IsInternal(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.Internal
|
|
}
|
|
func IsUnavailable(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.Unavailable
|
|
}
|
|
func IsDataLoss(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.DataLoss
|
|
}
|
|
func IsUnauthenticated(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return status.Convert(err).Code() == codes.Unauthenticated
|
|
}
|
|
|
|
func IsContextCanceled(err error) bool {
|
|
err = Unwrap(err)
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return strings.Contains(err.Error(), context.Canceled.Error())
|
|
}
|
|
|
|
func IsContextDeadlineExceeded(err error) bool {
|
|
err = Unwrap(err)
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return strings.Contains(err.Error(), context.DeadlineExceeded.Error())
|
|
}
|
|
|
|
func Unwrap(err error) error {
|
|
s, ok := status.FromError(err)
|
|
if s == nil {
|
|
return nil
|
|
}
|
|
if ok {
|
|
return errors.New(s.Message())
|
|
}
|
|
return err
|
|
}
|