-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
73 lines (61 loc) · 1.19 KB
/
main.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
package main
import (
"context"
"log"
"os"
"runtime/trace"
"sync"
)
const CONSUMERS_COUNT = 10
func main() {
f, err := os.Create("trace.out")
if err != nil {
log.Fatal(err)
}
defer func() {
if err := f.Close(); err != nil {
log.Fatal(err)
}
}()
if err := trace.Start(f); err != nil {
log.Fatal(err)
}
defer trace.Stop()
process()
}
func process() {
ctx, task := trace.NewTask(context.Background(), "process")
defer task.End()
var wg sync.WaitGroup
ch := make(chan int, 10)
// 1 ~ 100 までの整数を channel に送信する
go produce(100, ch, ctx)
// channel からデータを受信する
for i := 0; i < CONSUMERS_COUNT; i++ {
i := i
wg.Add(1)
go consume(i, ch, &wg, ctx)
}
wg.Wait()
}
func produce(num int, ch chan int, ctx context.Context) {
defer trace.StartRegion(ctx, "produce").End()
var pg sync.WaitGroup
defer close(ch)
for i := 1; i <= num; i++ {
i := i
pg.Add(1)
go func() {
defer pg.Done()
ch <- i
}()
}
pg.Wait()
}
func consume(idx int, ch chan int, wg *sync.WaitGroup, ctx context.Context) {
defer trace.StartRegion(ctx, "consume").End()
defer wg.Done()
for i := range ch {
log.Printf("#%d: consumed %d\n", idx, i)
}
}