Ants 的僵尸案例 :)

偶尔看见一个不错的协程池实现叫做 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 而占用过多资源。