Ants 的僵尸案例 :)

偶尔看见一个不错的协程池实现,觉得不错,但是官方例子有些冗余,所以自己写了个超简单的应用场景。
如果想要了解实现原理的可以点击【这里】,原作者已经写得很清楚了。

栗子

场景是–我们要处决僵尸,但是只有3把电椅。
当然,最主要的还是要有僵尸给我们盘,所以我们有两个方法:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
var tunnel = make(chan string, 1) // 僵尸隧道

// 处决僵尸
func ExecuteZombie(i interface{}) {
fmt.Printf("正在处决僵尸 %s 号,还有5秒钟....\n", i.(string))
time.Sleep(5*time.Second)
fmt.Printf(":) %s 玩完了,下一个\n-----------------\n", i.(string))
}

// 欢迎僵尸
func IncomingZombie() {
for i := 0; i < 10; i++ {
tunnel <- strconv.Itoa(i)
}
}

这一波大概会来10个左右的僵尸。
处理一个僵尸大概5秒钟。我们准备了三把电椅:

1
2
3
4
5
6
7
8
9
10
11
12
13
func main() {
go IncomingZombie()

chairPool, _ := ants.NewPoolWithFunc(3, ExecuteZombie) // 电椅就位
defer chairPool.Release() // 结束以后释放电椅

for {
select {
case a := <-tunnel:
go chairPool.Serve(a) // 来一个电一个
}
}
}

合起来就是

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
package main

import (
"fmt"
"strconv"
"time"

"github.com/panjf2000/ants"
)

var tunnel = make(chan string, 1)

func main() {
go IncomingZombie()

chairPool, _ := ants.NewPoolWithFunc(3, ExecuteZombie) // 声明有几把电椅
defer chairPool.Release()

for {
select {
case a := <-tunnel:
go chairPool.Serve(a)
}
}
}

// 处决僵尸
func ExecuteZombie(i interface{}) {
fmt.Printf("正在处决僵尸 %s 号,还有5秒钟....\n", i.(string))
time.Sleep(5*time.Second)
fmt.Printf(":) %s 玩完了,下一个\n-----------------\n", i.(string))
}

// 僵尸不断进来
func IncomingZombie() {
for i := 0; i < 10; i++ {
tunnel <- strconv.Itoa(i)
}
}

执行看看…

按照我们电椅的数量 ants.NewPoolWithFunc(3, ExecuteZombie)
你会发现,一次只会同步处决3枚僵尸。之后进来的都是出于堵塞的状态。

如果我们把椅子调整到100呢?
毫无疑问,全部一下子都处决了。

这种写法,最常用的模式就是阻塞等待队列消息的模型,而且对实时性要求不高。
同时你有不想因为过多的并发 routine 而占用过多资源。