Ants 的僵尸案例 :)

最新推荐文章于 2023-08-08 09:58:31 发布

cookedsteak

最新推荐文章于 2023-08-08 09:58:31 发布

阅读量472

点赞数

分类专栏：技术文章标签： ants golang 协程后端 go

本文链接：https://blog.youkuaiyun.com/cookedsteak/article/details/90690930

版权

技术专栏收录该内容

9 篇文章

订阅专栏

本文通过一个僵尸处决不超过3个电椅的场景，介绍了如何使用 Go 语言的 ants 协程池库进行简单应用。通过示例展示了 ants 如何控制并发数，避免资源过度占用，并在资源不足时进行阻塞等待。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

栗子

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

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秒钟。我们准备了三把电椅：

func main() {
	go IncomingZombie()

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

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

合起来就是

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 而占用过多资源。