74f76c75d5
This refactors a bit of the http handler code. Moves error handling for handler functions to a generic handler. This also has a side-effect of being able to propagate errors from the provider to send the correct status code, provided the error type matches a pre-defined interface.
86 lines
3.0 KiB
Go
86 lines
3.0 KiB
Go
package status
|
|
|
|
import (
|
|
"github.com/cpuguy83/strongerrors"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// FromGRPC returns an error class from the provided GPRC status
|
|
// If the status is nil or OK, this will return nil
|
|
// nolint: gocyclo
|
|
func FromGRPC(s *status.Status) error {
|
|
if s == nil || s.Code() == codes.OK {
|
|
return nil
|
|
}
|
|
|
|
switch s.Code() {
|
|
case codes.InvalidArgument:
|
|
return strongerrors.InvalidArgument(s.Err())
|
|
case codes.NotFound:
|
|
return strongerrors.NotFound(s.Err())
|
|
case codes.Unimplemented:
|
|
return strongerrors.NotImplemented(s.Err())
|
|
case codes.DeadlineExceeded:
|
|
return strongerrors.Deadline(s.Err())
|
|
case codes.Canceled:
|
|
return strongerrors.Cancelled(s.Err())
|
|
case codes.AlreadyExists:
|
|
return strongerrors.AlreadyExists(s.Err())
|
|
case codes.PermissionDenied:
|
|
return strongerrors.Unauthorized(s.Err())
|
|
case codes.Unauthenticated:
|
|
return strongerrors.Unauthenticated(s.Err())
|
|
// TODO(cpuguy83): consider more granular errors for these cases
|
|
case codes.FailedPrecondition, codes.Aborted, codes.Unavailable, codes.OutOfRange:
|
|
return strongerrors.Conflict(s.Err())
|
|
case codes.ResourceExhausted:
|
|
return strongerrors.Exhausted(s.Err())
|
|
case codes.DataLoss:
|
|
return strongerrors.DataLoss(s.Err())
|
|
default:
|
|
return strongerrors.Unknown(s.Err())
|
|
}
|
|
}
|
|
|
|
// ToGRPC takes the passed in error and converts it to a GRPC status error
|
|
// If the passed in error is already a gprc status error, then it is returned unmodified
|
|
// If the passed in error is nil, then a nil error is returned.
|
|
// nolint: gocyclo
|
|
func ToGRPC(err error) error {
|
|
if _, ok := status.FromError(err); ok {
|
|
return err
|
|
}
|
|
|
|
switch {
|
|
case strongerrors.IsNotFound(err):
|
|
return status.Error(codes.NotFound, err.Error())
|
|
case strongerrors.IsConflict(err), strongerrors.IsNotModified(err):
|
|
return status.Error(codes.FailedPrecondition, err.Error())
|
|
case strongerrors.IsInvalidArgument(err):
|
|
return status.Error(codes.InvalidArgument, err.Error())
|
|
case strongerrors.IsAlreadyExists(err):
|
|
return status.Error(codes.AlreadyExists, err.Error())
|
|
case strongerrors.IsCancelled(err):
|
|
return status.Error(codes.Canceled, err.Error())
|
|
case strongerrors.IsDeadline(err):
|
|
return status.Error(codes.DeadlineExceeded, err.Error())
|
|
case strongerrors.IsUnauthorized(err):
|
|
return status.Error(codes.PermissionDenied, err.Error())
|
|
case strongerrors.IsUnauthenticated(err):
|
|
return status.Error(codes.Unauthenticated, err.Error())
|
|
case strongerrors.IsForbidden(err), strongerrors.IsNotImplemented(err):
|
|
return status.Error(codes.Unimplemented, err.Error())
|
|
case strongerrors.IsExhausted(err):
|
|
return status.Error(codes.ResourceExhausted, err.Error())
|
|
case strongerrors.IsDataLoss(err):
|
|
return status.Error(codes.DataLoss, err.Error())
|
|
case strongerrors.IsSystem(err):
|
|
return status.Error(codes.Internal, err.Error())
|
|
case strongerrors.IsUnavailable(err):
|
|
return status.Error(codes.Unavailable, err.Error())
|
|
default:
|
|
return status.Error(codes.Unknown, err.Error())
|
|
}
|
|
}
|