blob: 50e7e96430162d1d5a2b0c1821d1ba6d5117e1ec (
plain)
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
|
package fanout_test
import (
"context"
"fmt"
"sort"
"sync"
"git.sangtran.io.vn/quacker.git/fanout"
)
func ExampleNewFanOut() {
fo, _ := fanout.NewFanOut[int](
fanout.WithWorkerCount(2),
fanout.WithBufferSize(10),
fanout.WithStrategy(fanout.RoundRobin),
)
input := make(chan int, 4)
for i := 1; i <= 4; i++ {
input <- i
}
close(input)
fo.Run(context.Background(), input)
var mu sync.Mutex
var all []int
var wg sync.WaitGroup
for _, ch := range fo.Outputs() {
wg.Go(func() {
for v := range ch {
mu.Lock()
all = append(all, v)
mu.Unlock()
}
})
}
wg.Wait()
sort.Ints(all)
fmt.Println(all)
// Output: [1 2 3 4]
}
|
