You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
119 lines
2.8 KiB
119 lines
2.8 KiB
package errors
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/go-kratos/kratos/v2/internal/httputil"
|
|
"google.golang.org/genproto/googleapis/rpc/errdetails"
|
|
"google.golang.org/grpc/status"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
const (
|
|
// UnknownCode is unknown code for error info.
|
|
UnknownCode = 500
|
|
// UnknownReason is unknown reason for error info.
|
|
UnknownReason = ""
|
|
)
|
|
|
|
//go:generate protoc -I. --go_out=paths=source_relative:. errors.proto
|
|
|
|
func (e *Error) Error() string {
|
|
return fmt.Sprintf("error: code = %d reason = %s message = %s metadata = %v", e.Code, e.Reason, e.Message, e.Metadata)
|
|
}
|
|
|
|
// StatusCode return an HTTP error code.
|
|
func (e *Error) StatusCode() int {
|
|
return int(e.Code)
|
|
}
|
|
|
|
// GRPCStatus returns the Status represented by se.
|
|
func (e *Error) GRPCStatus() *status.Status {
|
|
s, _ := status.New(httputil.GRPCCodeFromStatus(e.StatusCode()), e.Message).
|
|
WithDetails(&errdetails.ErrorInfo{
|
|
Reason: e.Reason,
|
|
Metadata: e.Metadata,
|
|
})
|
|
return s
|
|
}
|
|
|
|
// Is matches each error in the chain with the target value.
|
|
func (e *Error) Is(err error) bool {
|
|
if se := new(Error); errors.As(err, &se) {
|
|
return se.Reason == e.Reason
|
|
}
|
|
return false
|
|
}
|
|
|
|
// WithMetadata with an MD formed by the mapping of key, value.
|
|
func (e *Error) WithMetadata(md map[string]string) *Error {
|
|
err := proto.Clone(e).(*Error)
|
|
err.Metadata = md
|
|
return err
|
|
}
|
|
|
|
// New returns an error object for the code, message.
|
|
func New(code int, reason, message string) *Error {
|
|
return &Error{
|
|
Code: int32(code),
|
|
Message: message,
|
|
Reason: reason,
|
|
}
|
|
}
|
|
|
|
// Newf New(code fmt.Sprintf(format, a...))
|
|
func Newf(code int, reason, format string, a ...interface{}) *Error {
|
|
return New(code, reason, fmt.Sprintf(format, a...))
|
|
}
|
|
|
|
// Errorf returns an error object for the code, message and error info.
|
|
func Errorf(code int, reason, format string, a ...interface{}) error {
|
|
return New(code, reason, fmt.Sprintf(format, a...))
|
|
}
|
|
|
|
// Code returns the code for a particular error.
|
|
// It supports wrapped errors.
|
|
func Code(err error) int {
|
|
if err == nil {
|
|
return 200
|
|
}
|
|
if se := FromError(err); err != nil {
|
|
return int(se.Code)
|
|
}
|
|
return UnknownCode
|
|
}
|
|
|
|
// Reason returns the reason for a particular error.
|
|
// It supports wrapped errors.
|
|
func Reason(err error) string {
|
|
if se := FromError(err); err != nil {
|
|
return se.Reason
|
|
}
|
|
return UnknownReason
|
|
}
|
|
|
|
// FromError try to convert an error to *Error.
|
|
// It supports wrapped errors.
|
|
func FromError(err error) *Error {
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
if se := new(Error); errors.As(err, &se) {
|
|
return se
|
|
}
|
|
gs, ok := status.FromError(err)
|
|
if ok {
|
|
for _, detail := range gs.Details() {
|
|
switch d := detail.(type) {
|
|
case *errdetails.ErrorInfo:
|
|
return New(
|
|
httputil.StatusFromGRPCCode(gs.Code()),
|
|
d.Reason,
|
|
gs.Message(),
|
|
).WithMetadata(d.Metadata)
|
|
}
|
|
}
|
|
}
|
|
return New(UnknownCode, UnknownReason, err.Error())
|
|
}
|
|
|