/
guard.go
109 lines (95 loc) · 2.28 KB
/
guard.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
102
103
104
105
106
107
108
109
package xx
import (
"context"
"fmt"
)
// Ign set err to nil if one of fn returns true
func Ign(err *error, fns ...F11[error, bool]) error {
if *err == nil {
return *err
}
for _, fn := range fns {
if fn(*err) {
*err = nil
return *err
}
}
return *err
}
// Guard recover from panic and set err
func Guard(err *error) {
if r := recover(); r != nil {
if re, ok := r.(error); ok {
*err = re
} else {
*err = fmt.Errorf("panic: %v", r)
}
}
}
// MustCtx panic ctx.Err() if not nil
func MustCtx(ctx context.Context) {
if ctx.Err() != nil {
panic(ctx.Err())
}
}
// Must0 panic err if not nil
func Must0(err error) {
if err != nil {
panic(err)
}
}
// Must1 panic err if not nil, else return remaining values
func Must1[T any](v T, err error) T {
if err == nil {
return v
} else {
panic(err)
}
}
// Must alias to Must1
func Must[T any](v T, err error) T {
return Must1(v, err)
}
// __BEGIN_OF_GENERATION__
// Must2 panic err if not nil, else return remaining values
func Must2[T1 any, T2 any](v1 T1, v2 T2, err error) (T1, T2) {
if err != nil {
panic(err)
}
return v1, v2
}
// Must3 panic err if not nil, else return remaining values
func Must3[T1 any, T2 any, T3 any](v1 T1, v2 T2, v3 T3, err error) (T1, T2, T3) {
if err != nil {
panic(err)
}
return v1, v2, v3
}
// Must4 panic err if not nil, else return remaining values
func Must4[T1 any, T2 any, T3 any, T4 any](v1 T1, v2 T2, v3 T3, v4 T4, err error) (T1, T2, T3, T4) {
if err != nil {
panic(err)
}
return v1, v2, v3, v4
}
// Must5 panic err if not nil, else return remaining values
func Must5[T1 any, T2 any, T3 any, T4 any, T5 any](v1 T1, v2 T2, v3 T3, v4 T4, v5 T5, err error) (T1, T2, T3, T4, T5) {
if err != nil {
panic(err)
}
return v1, v2, v3, v4, v5
}
// Must6 panic err if not nil, else return remaining values
func Must6[T1 any, T2 any, T3 any, T4 any, T5 any, T6 any](v1 T1, v2 T2, v3 T3, v4 T4, v5 T5, v6 T6, err error) (T1, T2, T3, T4, T5, T6) {
if err != nil {
panic(err)
}
return v1, v2, v3, v4, v5, v6
}
// Must7 panic err if not nil, else return remaining values
func Must7[T1 any, T2 any, T3 any, T4 any, T5 any, T6 any, T7 any](v1 T1, v2 T2, v3 T3, v4 T4, v5 T5, v6 T6, v7 T7, err error) (T1, T2, T3, T4, T5, T6, T7) {
if err != nil {
panic(err)
}
return v1, v2, v3, v4, v5, v6, v7
}