From 13e53f0c88083b9d9b32ae6a458c751a00ff768b Mon Sep 17 00:00:00 2001 From: jdl Date: Tue, 5 Dec 2023 11:24:03 +0100 Subject: [PATCH] Updated error type --- lib/errs/error.go | 45 +++++++++++++++++++++++---------------------- lib/errs/fmt.go | 8 ++++---- 2 files changed, 27 insertions(+), 26 deletions(-) diff --git a/lib/errs/error.go b/lib/errs/error.go index 8857194..ecb2646 100644 --- a/lib/errs/error.go +++ b/lib/errs/error.go @@ -8,26 +8,27 @@ import ( ) type Error struct { - msg string - code int64 + Code int64 + Msg string + StackTrace string + collection string index string - stackTrace string err error // Wrapped error } func NewErr(code int64, msg string) *Error { return &Error{ - msg: msg, - code: code, + Code: code, + Msg: msg, } } func (e *Error) Error() string { if e.collection != "" || e.index != "" { - return fmt.Sprintf(`[%d] (%s/%s) %s`, e.code, e.collection, e.index, e.msg) + return fmt.Sprintf(`[%d] (%s/%s) %s`, e.Code, e.collection, e.index, e.Msg) } else { - return fmt.Sprintf("[%d] %s", e.code, e.msg) + return fmt.Sprintf("[%d] %s", e.Code, e.Msg) } } @@ -36,11 +37,15 @@ func (e *Error) Is(rhs error) bool { if !ok { return false } - return e.code == e2.code + return e.Code == e2.Code +} + +func (e *Error) Unwrap() error { + return e.err } func (e *Error) WithErr(err error) *Error { - if e2, ok := err.(*Error); ok && e2.code == e.code { + if e2, ok := err.(*Error); ok && e2.Code == e.Code { return e2 } @@ -49,15 +54,11 @@ func (e *Error) WithErr(err error) *Error { return e2 } -func (e *Error) Unwrap() error { - return e.err -} - func (e *Error) WithMsg(msg string, args ...any) *Error { err := *e - err.msg += ": " + fmt.Sprintf(msg, args...) - if len(err.stackTrace) == 0 { - err.stackTrace = string(debug.Stack()) + err.Msg += ": " + fmt.Sprintf(msg, args...) + if len(err.StackTrace) == 0 { + err.StackTrace = string(debug.Stack()) } return &err } @@ -75,16 +76,16 @@ func (e *Error) WithIndex(s string) *Error { } func (e *Error) msgTruncacted() string { - if len(e.msg) > 255 { - return e.msg[:255] + if len(e.Msg) > 255 { + return e.Msg[:255] } - return e.msg + return e.Msg } func (e *Error) Write(w io.Writer) error { msg := e.msgTruncacted() - if err := binary.Write(w, binary.LittleEndian, e.code); err != nil { + if err := binary.Write(w, binary.LittleEndian, e.Code); err != nil { return IO.WithErr(err) } @@ -100,7 +101,7 @@ func (e *Error) Read(r io.Reader) error { size uint8 ) - if err := binary.Read(r, binary.LittleEndian, &e.code); err != nil { + if err := binary.Read(r, binary.LittleEndian, &e.Code); err != nil { return IO.WithErr(err) } @@ -113,6 +114,6 @@ func (e *Error) Read(r io.Reader) error { return IO.WithErr(err) } - e.msg = string(msgBuf) + e.Msg = string(msgBuf) return nil } diff --git a/lib/errs/fmt.go b/lib/errs/fmt.go index fa43dfc..bacffd0 100644 --- a/lib/errs/fmt.go +++ b/lib/errs/fmt.go @@ -10,12 +10,12 @@ func FmtDetails(err error) string { var s string if e.collection != "" || e.index != "" { - s = fmt.Sprintf(`[%d] (%s/%s) %s`, e.code, e.collection, e.index, e.msg) + s = fmt.Sprintf(`[%d] (%s/%s) %s`, e.Code, e.collection, e.index, e.Msg) } else { - s = fmt.Sprintf("[%d] %s", e.code, e.msg) + s = fmt.Sprintf("[%d] %s", e.Code, e.Msg) } - if len(e.stackTrace) != 0 { - s += "\n\nStack Trace:\n" + e.stackTrace + "\n" + if len(e.StackTrace) != 0 { + s += "\n\nStack Trace:\n" + e.StackTrace + "\n" } return s