-
Notifications
You must be signed in to change notification settings - Fork 0
/
invoke.go
149 lines (119 loc) · 3.24 KB
/
invoke.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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
package fo
import (
"context"
"sync"
)
func invoke[R any](ctx context.Context, fn func() (R, error)) (R, error) {
var res R
var err error
resChan := make(chan struct{})
go func() {
res, err = fn()
resChan <- struct{}{}
}()
var wg sync.WaitGroup
wg.Add(1)
go func() {
select {
case <-ctx.Done():
err = ctx.Err()
case <-resChan:
}
wg.Done()
}()
wg.Wait()
return res, err
}
// Invoke0 has the same behavior as Invoke but without return value.
func Invoke0(ctx context.Context, fn func() error) error {
_, err := invoke(ctx, func() (any, error) {
return nil, fn()
})
return err
}
// Invoke invokes the callback function and enables to control the
// context of the callback function with 1 return value.
func Invoke[R1 any](ctx context.Context, fn func() (R1, error)) (R1, error) {
return Invoke1(ctx, fn)
}
// Invoke1 is an alias of Invoke.
func Invoke1[R1 any](ctx context.Context, fn func() (R1, error)) (R1, error) {
type result struct {
r1 R1
}
res, err := invoke(ctx, func() (result, error) {
r1, err := fn()
return result{r1: r1}, err
})
return res.r1, err
}
// Invoke2 has the same behavior as Invoke but with 2 return values.
func Invoke2[R1 any, R2 any](ctx context.Context, fn func() (R1, R2, error)) (R1, R2, error) {
type result struct {
r1 R1
r2 R2
}
res, err := invoke(ctx, func() (result, error) {
r1, r2, err := fn()
return result{r1: r1, r2: r2}, err
})
return res.r1, res.r2, err
}
// Invoke3 has the same behavior as Invoke but with 3 return values.
func Invoke3[R1 any, R2 any, R3 any](ctx context.Context, fn func() (R1, R2, R3, error)) (R1, R2, R3, error) {
type result struct {
r1 R1
r2 R2
r3 R3
}
res, err := invoke(ctx, func() (result, error) {
r1, r2, r3, err := fn()
return result{r1: r1, r2: r2, r3: r3}, err
})
return res.r1, res.r2, res.r3, err
}
// Invoke4 has the same behavior as Invoke but with 4 return values.
func Invoke4[R1 any, R2 any, R3 any, R4 any](ctx context.Context, fn func() (R1, R2, R3, R4, error)) (R1, R2, R3, R4, error) {
type result struct {
r1 R1
r2 R2
r3 R3
r4 R4
}
res, err := invoke(ctx, func() (result, error) {
r1, r2, r3, r4, err := fn()
return result{r1: r1, r2: r2, r3: r3, r4: r4}, err
})
return res.r1, res.r2, res.r3, res.r4, err
}
// Invoke5 has the same behavior as Invoke but with 5 return values.
func Invoke5[R1 any, R2 any, R3 any, R4 any, R5 any](ctx context.Context, fn func() (R1, R2, R3, R4, R5, error)) (R1, R2, R3, R4, R5, error) {
type result struct {
r1 R1
r2 R2
r3 R3
r4 R4
r5 R5
}
res, err := invoke(ctx, func() (result, error) {
r1, r2, r3, r4, r5, err := fn()
return result{r1: r1, r2: r2, r3: r3, r4: r4, r5: r5}, err
})
return res.r1, res.r2, res.r3, res.r4, res.r5, err
}
// Invoke6 has the same behavior as Invoke but with 6 return values.
func Invoke6[R1 any, R2 any, R3 any, R4 any, R5 any, R6 any](ctx context.Context, fn func() (R1, R2, R3, R4, R5, R6, error)) (R1, R2, R3, R4, R5, R6, error) {
type result struct {
r1 R1
r2 R2
r3 R3
r4 R4
r5 R5
r6 R6
}
res, err := invoke(ctx, func() (result, error) {
r1, r2, r3, r4, r5, r6, err := fn()
return result{r1: r1, r2: r2, r3: r3, r4: r4, r5: r5, r6: r6}, err
})
return res.r1, res.r2, res.r3, res.r4, res.r5, res.r6, err
}