utils.go 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. package errors
  2. import (
  3. "errors"
  4. "strings"
  5. )
  6. func NewClass(code string, msgList ...string) WTErrorClass {
  7. var msg string
  8. if len(msgList) == 0 {
  9. msg = strings.Replace(code, "_", " ", -1)
  10. } else {
  11. msg = msgList[0]
  12. }
  13. return &wtErrorClass{
  14. msg: msg,
  15. code: code,
  16. base: false,
  17. }
  18. }
  19. func newBaseClass(code string, msgList ...string) WTErrorClass {
  20. var msg string
  21. if len(msgList) == 0 {
  22. msg = strings.Replace(code, "_", " ", -1)
  23. } else {
  24. msg = msgList[0]
  25. }
  26. return &wtErrorClass{
  27. msg: msg,
  28. code: code,
  29. base: true,
  30. }
  31. }
  32. func Is(err error, target any) bool {
  33. var wtErr WTError
  34. var normalErr error
  35. if errors.As(err, &wtErr) {
  36. errClass, isClass := target.(WTErrorClass)
  37. if isClass {
  38. return errClass.Code() == wtErr.Code()
  39. }
  40. wtTarget, isTarget := target.(WTError)
  41. if !isTarget {
  42. return false
  43. }
  44. return errors.Is(err, wtTarget)
  45. } else if errors.As(err, &normalErr) {
  46. targetErr, targetIsErr := target.(error)
  47. if !targetIsErr {
  48. return false
  49. }
  50. return errors.Is(err, targetErr)
  51. }
  52. return false
  53. }
  54. func As(err error, target any) bool {
  55. return errors.As(err, target)
  56. }