go-errors/errors.go at master · deweppro/go-errors · GitHub

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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package errors

import (
	e "errors"
	"fmt"
)

type errMessage struct {
	cause   error
	message string
	trace   string
}

func New(message string) error {
	return &errMessage{message: message}
}

func (v *errMessage) Error() string {
	switch true {
	case len(v.message) > 0 && v.cause != nil:
		return v.message + ": " + v.cause.Error() + v.trace
	case v.cause != nil:
		return v.cause.Error() + v.trace
	}
	return v.message + v.trace
}

func (v *errMessage) Cause() error {
	return v.cause
}

func (v *errMessage) Unwrap() error {
	return v.cause
}

func (v *errMessage) WithTrace() {
	v.trace = tracing()
}

///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

func WrapMessage(cause error, message string, args ...interface{}) error {
	if cause == nil {
		return nil
	}
	var err0 *errMessage
	if len(args) == 0 {
		err0 = &errMessage{
			cause:   cause,
			message: message,
		}
	} else {
		err0 = &errMessage{
			cause:   cause,
			message: fmt.Sprintf(message, args...),
		}
	}
	err0.WithTrace()
	return err0
}

func Wrap(msg ...error) error {
	if len(msg) == 0 {
		return nil
	}
	var err0 error
	for _, v := range msg {
		if v == nil {
			continue
		}
		if err0 == nil {
			err0 = &errMessage{cause: v}
			continue
		}
		err0 = &errMessage{
			cause:   v,
			message: err0.Error(),
		}
	}
	return err0
}

///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

type (
	IUnwrap interface {
		Unwrap() error
	}
	ICause interface {
		Cause() error
	}
)

func Unwrap(err error) error {
	if v, ok := err.(IUnwrap); ok {
		return v.Unwrap()
	}
	return nil
}

func Cause(err error) error {
	for err != nil {
		v, ok := err.(ICause)
		if !ok {
			break
		}
		err = v.Cause()
	}

	return nil
}

func Is(err, target error) bool {
	return e.Is(err, target)
}