-
Notifications
You must be signed in to change notification settings - Fork 16
/
error.go
84 lines (75 loc) · 2.08 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
package super
import (
"errors"
"sync"
)
var errorManagerInstance *errorManager
// RegError 通过错误码注册错误,返回错误的引用
func RegError(code int, message string) error {
if code == 0 {
return errors.New("error code can not be 0")
}
errorManagerInstance.mutex.Lock()
defer errorManagerInstance.mutex.Unlock()
if errorManagerInstance == nil {
errorManagerInstance = new(errorManager).init()
}
err := &ser{code: code, message: message}
errorManagerInstance.errorMapper[err] = code
return err
}
// RegErrorRef 通过错误码注册错误,返回错误的引用
// - 引用将会被重定向到注册的错误信息
func RegErrorRef(code int, message string, ref error) error {
if code == 0 {
return errors.New("error code can not be 0")
}
errorManagerInstance.mutex.Lock()
defer errorManagerInstance.mutex.Unlock()
if errorManagerInstance == nil {
errorManagerInstance = new(errorManager).init()
}
err := &ser{code: code, message: message}
errorManagerInstance.errorMapper[err] = code
errorManagerInstance.errorMapperRef[ref] = err
return ref
}
// GetError 通过错误引用获取错误码和真实错误信息,如果错误不存在则返回 0,如果错误引用不存在则返回原本的错误
func GetError(err error) (int, error) {
unw := errors.Unwrap(err)
if unw == nil {
unw = err
}
errorManagerInstance.mutex.Lock()
defer errorManagerInstance.mutex.Unlock()
if ref, exist := errorManagerInstance.errorMapperRef[unw]; exist {
//err = fmt.Errorf("%w : %s", ref, err.Error())
err = ref
}
unw = errors.Unwrap(err)
if unw == nil {
unw = err
}
code, exist := errorManagerInstance.errorMapper[unw]
if !exist {
return 0, errors.New("error not found")
}
return code, err
}
type ser struct {
code int
message string
}
func (slf *ser) Error() string {
return slf.message
}
type errorManager struct {
errorMapper map[error]int
errorMapperRef map[error]error
mutex sync.Mutex
}
func (slf *errorManager) init() *errorManager {
slf.errorMapper = make(map[error]int)
slf.errorMapperRef = make(map[error]error)
return slf
}