generated from dogmatiq/template-go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
executor.go
68 lines (55 loc) · 1017 Bytes
/
executor.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
package discovery
import (
"context"
"sync"
)
// executor provides shared logic to TargetExecutor and ClientExecutor.
type executor struct {
m sync.Mutex
tasks map[interface{}]task
}
type task struct {
Cancel context.CancelFunc
Done chan struct{}
}
func (e *executor) start(
parent context.Context,
key interface{},
fn func(context.Context),
) {
e.m.Lock()
defer e.m.Unlock()
if e.tasks == nil {
e.tasks = map[interface{}]task{}
} else if _, ok := e.tasks[key]; ok {
return
}
if parent == nil {
parent = context.Background()
}
ctx, cancel := context.WithCancel(parent)
done := make(chan struct{})
e.tasks[key] = task{
cancel,
done,
}
go func() {
defer close(done)
fn(ctx)
}()
}
func (e *executor) stop(key interface{}) {
if task, ok := e.remove(key); ok {
task.Cancel()
<-task.Done
}
}
func (e *executor) remove(key interface{}) (task, bool) {
e.m.Lock()
defer e.m.Unlock()
task, ok := e.tasks[key]
if ok {
delete(e.tasks, key)
}
return task, ok
}