/
main.go
90 lines (79 loc) · 1.56 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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package main
import (
"fmt"
"math/rand"
"sync"
"time"
)
// 使用goroutine和channel实现计算int64随机数和的程序
// 1. 开启goroutine循环生成随机数发送到jobChan
// 2. 开启24个goroutine从jobChan中取出随机数计算各位数的和,结果发送resultChan
// 3. 从resultChan取出并打印
// Job 随机数
type Job struct {
value int64
}
// Result 计算各位数的和
type Result struct {
job *Job
sum int64
}
var wg sync.WaitGroup
var once sync.Once
var jobChan = make(chan *Job, 100)
var resultChan = make(chan *Result, 100)
func init() {
rand.Seed(time.Now().UnixNano())
}
func generate(data chan<- *Job) {
defer wg.Done()
defer close(data) //避免deadlock
var i = 0
for {
x := rand.Int63()
newJob := &Job{
value: x,
}
data <- newJob
time.Sleep(time.Millisecond * 500)
i++
// if i > 5 { //避免deadlock
// break
// }
}
}
func calc(data <-chan *Job, resultChan chan<- *Result) {
defer wg.Done()
defer once.Do(func() { //避免deadlock
close(resultChan)
})
for {
job, ok := <-data
if !ok { //避免deadlock
break
}
n := job.value
var sum int64
for n > 0 {
sum += n % 10
n /= 10
}
newResult := &Result{
job: job,
sum: sum,
}
resultChan <- newResult
}
}
func main() {
wg.Add(1)
go generate(jobChan)
wg.Add(24)
for i := 0; i < 24; i++ {
go calc(jobChan, resultChan)
}
for result := range resultChan {
fmt.Printf("value: %d sum:%d\n", result.job.value, result.sum)
}
wg.Wait()
}