forked from neoxelox/kit
-
Notifications
You must be signed in to change notification settings - Fork 0
/
error.go
101 lines (78 loc) · 1.93 KB
/
error.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package kit
import (
"fmt"
"github.com/cockroachdb/errors"
)
// TODO: Wrap and WrapAs just call WithDepth +1
// TODO: dedup error messages
// TODO: simplify all of this, maybe this is overkill that adds nothing
type Error struct {
inner error
identifier string
message string
}
func NewError(message string) func() *Error {
return func() *Error {
return &Error{
inner: errors.NewWithDepth(1, message),
identifier: message,
message: message,
}
}
}
func (self *Error) Wrap(err error) *Error {
if err != nil {
self.inner = errors.WrapWithDepth(1, err, self.message)
}
return self
}
func (self *Error) WrapWithDepth(depth int, err error) *Error {
if err != nil {
self.inner = errors.WrapWithDepth(depth+1, err, self.message)
}
return self
}
func (self *Error) WrapAs(err error) *Error {
if err != nil {
if other, ok := err.(*Error); ok {
self.identifier = other.identifier
} else {
self.identifier = err.Error()
}
self.inner = errors.WrapWithDepth(1, err, self.message)
}
return self
}
func (self *Error) WrapAsWithDepth(depth int, err error) *Error {
if err != nil {
if other, ok := err.(*Error); ok {
self.identifier = other.identifier
} else {
self.identifier = err.Error()
}
self.inner = errors.WrapWithDepth(depth+1, err, self.message)
}
return self
}
func (self *Error) With(message string) *Error {
self.message = self.message + ": " + message
self.inner = errors.NewWithDepth(1, self.message)
return self
}
func (self *Error) Withf(message string, args ...any) *Error {
self.message = self.message + ": " + fmt.Sprintf(message, args...)
self.inner = errors.NewWithDepth(1, self.message)
return self
}
func (self Error) Error() string {
return self.inner.Error()
}
func (self Error) Unwrap() error {
return self.inner
}
func (self Error) Is(err error) bool {
if other, ok := err.(*Error); ok {
return self.identifier == other.identifier
}
return false
}