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