题目来源:京东
答案:古尔班通
该问题适合使用pipeline模式(pipeline是一系列将数据输入,执行操作并将数据传回的系统,我们称这些操作都是pipeline的一个stage),代码如下:
package main
import (
"fmt"
"math/rand"
)
func main() {
// 随机数生成器
producer := func(num int, done <-chan struct{}) <-chan int {
produceStream := make(chan int)
go func() {
defer close(produceStream)
for i := 0; i < num; i++ {
select {
case <-done:
return
case produceStream <- rand.Intn(99):
}
}
}()
return produceStream
}
// 随机数倍增器
consumer := func(done <-chan struct{}, produceStream <-chan int) <-chan int {
consumerStream := make(chan int)
go func() {
defer close(consumerStream)
for v := range produceStream {
select {
case <-done:
return
case consumerStream <- v * 2:
}
}
}()
return consumerStream
}
done := make(chan struct{})
defer close(done)
produceStream := producer(10, done)
consumerStream := consumer(done, produceStream)
for output := range consumerStream {
fmt.Println("Output: ", output)
}
}
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
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