-
Notifications
You must be signed in to change notification settings - Fork 0
/
pipeline.go
48 lines (37 loc) · 910 Bytes
/
pipeline.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
package mediator
import (
"context"
)
type Pipeline struct {
behaviours []BehaviorFunc
pipeline PipelineFunc
handlers map[string]RequestHandler
}
func New() *Pipeline {
return &Pipeline{
handlers: make(map[string]RequestHandler),
}
}
func (p *Pipeline) UseBehaviour(behaviour PipelineBehaviour) Builder {
return p.Use(behaviour.Process)
}
func (p *Pipeline) Use(call func(context.Context, Message, NextFunc) (interface{}, error)) Builder {
p.behaviours = append(p.behaviours, call)
return p
}
func (p *Pipeline) RegisterHandler(req Message, h RequestHandler) Builder {
key := req.Key()
p.handlers[key] = h
return p
}
func (p *Pipeline) Build() (*Mediator, error) {
m := newMediator(*p)
reverseApply(p.behaviours, m.pipe)
return m, nil
}
func reverseApply(behaviours []BehaviorFunc,
fn func(BehaviorFunc)) {
for i := len(behaviours) - 1; i >= 0; i-- {
fn(behaviours[i])
}
}