forked from juju/utils
/
parallel.go
92 lines (83 loc) · 2.02 KB
/
parallel.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
// Copyright 2013 Canonical Ltd.
// Licensed under the LGPLv3, see LICENCE file for details.
// The parallel package provides utilities for running tasks
// concurrently.
package parallel
import (
"fmt"
"sync"
)
// Run represents a number of functions running concurrently.
type Run struct {
mu sync.Mutex
results chan Errors
max int
running int
work chan func() error
}
// Errors holds any errors encountered during the parallel run.
type Errors []error
func (errs Errors) Error() string {
switch len(errs) {
case 0:
return "no error"
case 1:
return errs[0].Error()
}
return fmt.Sprintf("%s (and %d more)", errs[0].Error(), len(errs)-1)
}
// NewRun returns a new parallel instance. It provides a way of running
// functions concurrently while limiting the maximum number running at
// once to max.
func NewRun(max int) *Run {
if max < 1 {
panic("parameter max must be >= 1")
}
return &Run{
max: max,
results: make(chan Errors),
work: make(chan func() error),
}
}
// Do requests that r run f concurrently. If there are already the maximum
// number of functions running concurrently, it will block until one of them
// has completed. Do may itself be called concurrently, but may not be called
// concurrently with Wait.
func (r *Run) Do(f func() error) {
select {
case r.work <- f:
return
default:
}
r.mu.Lock()
if r.running < r.max {
r.running++
go r.runner()
}
r.mu.Unlock()
r.work <- f
}
// Wait marks the parallel instance as complete and waits for all the functions
// to complete. If any errors were encountered, it returns an Errors value
// describing all the errors in arbitrary order.
func (r *Run) Wait() error {
close(r.work)
var errs Errors
for i := 0; i < r.running; i++ {
errs = append(errs, <-r.results...)
}
if len(errs) == 0 {
return nil
}
// TODO(rog) sort errors by original order of Do request?
return errs
}
func (r *Run) runner() {
var errs Errors
for f := range r.work {
if err := f(); err != nil {
errs = append(errs, err)
}
}
r.results <- errs
}