forked from jrallison/go-workers
-
Notifications
You must be signed in to change notification settings - Fork 0
/
middleware.go
62 lines (50 loc) · 1.26 KB
/
middleware.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
package workers
type Action interface {
Call(queue string, message *Msg, next func() error) error
}
type Middlewares struct {
actions []Action
}
func (m *Middlewares) Append(action Action) {
m.actions = append(m.actions, action)
}
func (m *Middlewares) Prepend(action Action) {
actions := make([]Action, len(m.actions)+1)
actions[0] = action
copy(actions[1:], m.actions)
m.actions = actions
}
func (m *Middlewares) Equals(other *Middlewares) bool {
if len(m.actions) != len(other.actions) {
return false
}
for i := range m.actions {
if m.actions[i] != other.actions[i] {
return false
}
}
return true
}
func (m *Middlewares) AppendToCopy(mids []Action) *Middlewares {
return NewMiddleware(append(m.actions, mids...)...)
}
func (m *Middlewares) call(queue string, message *Msg, final func() error) error {
return continuation(m.actions, queue, message, final)()
}
func continuation(actions []Action, queue string, message *Msg, final func() error) func() error {
return func() (err error) {
if len(actions) > 0 {
err = actions[0].Call(
queue,
message,
continuation(actions[1:], queue, message, final),
)
return
} else {
return final()
}
}
}
func NewMiddleware(actions ...Action) *Middlewares {
return &Middlewares{actions}
}