Redmage/pkg/errs/query.go

55 lines
1.1 KiB
Go
Raw Normal View History

2024-04-06 21:11:22 +07:00
package errs
2024-04-26 10:51:09 +07:00
import (
"errors"
)
2024-04-06 21:11:22 +07:00
func FindCodeOrDefault(err error, def int) int {
2024-04-26 10:51:09 +07:00
if coder, ok := err.(interface{ GetCode() int }); ok {
code := coder.GetCode()
if code != 0 {
return code
}
}
for unwrap := errors.Unwrap(err); unwrap != nil; unwrap = errors.Unwrap(unwrap) {
if coder, ok := unwrap.(interface{ GetCode() int }); ok {
2024-04-06 21:11:22 +07:00
code := coder.GetCode()
if code != 0 {
return code
2024-04-06 21:11:22 +07:00
}
}
}
2024-04-06 21:11:22 +07:00
return def
}
func FindMessage(err error) string {
2024-04-26 10:51:09 +07:00
if messager, ok := err.(interface{ GetMessage() string }); ok {
message := messager.GetMessage()
if message != "" {
return message
}
}
for unwrap := errors.Unwrap(err); unwrap != nil; unwrap = errors.Unwrap(unwrap) {
if messager, ok := unwrap.(interface{ GetMessage() string }); ok {
message := messager.GetMessage()
if message != "" {
return message
}
}
}
return err.Error()
}
2024-04-24 22:26:04 +07:00
func HTTPMessage(err error) (code int, message string) {
code = FindCodeOrDefault(err, 500)
if code >= 500 {
return code, err.Error()
}
2024-04-26 10:51:09 +07:00
message = FindMessage(err)
return code, message
2024-04-24 22:26:04 +07:00
}