Golang高性能goroutine池管理库ants使用指南

ants是一个高性能且低成本的goroutine池实现，特别适合需要大量goroutine但希望控制资源消耗的场景。下面详细介绍ants的使用方法。

安装

go get -u github.com/panjf2000/ants/v2

基本使用

1. 创建goroutine池

package main

import (
	"fmt"
	"sync"
	"time"

	"github.com/panjf2000/ants/v2"
)

func demoFunc() {
	time.Sleep(1 * time.Second)
	fmt.Println("Hello World!")
}

func main() {
	defer ants.Release()

	runTimes := 1000

	// 使用公共池
	var wg sync.WaitGroup
	for i := 0; i < runTimes; i++ {
		wg.Add(1)
		ants.Submit(func() {
			demoFunc()
			wg.Done()
		})
	}
	wg.Wait()

	fmt.Printf("running goroutines: %d\n", ants.Running())
	fmt.Printf("total goroutines: %d\n", ants.Cap())
}

2. 自定义池大小

func main() {
	// 设置池容量为10
	p, _ := ants.NewPool(10)
	defer p.Release()

	for i := 0; i < 100; i++ {
		p.Submit(func() {
			demoFunc()
		})
	}

	// 获取当前正在运行的goroutine数量
	fmt.Printf("running goroutines: %d\n", p.Running())
}

高级特性

1. 带参数的函数

func demoFuncWithArgs(i int) {
	time.Sleep(1 * time.Second)
	fmt.Printf("Hello World! %d\n", i)
}

func main() {
	p, _ := ants.NewPool(10)
	defer p.Release()

	for i := 0; i < 100; i++ {
		// 使用闭包传递参数
		p.Submit(func() {
			demoFuncWithArgs(i)
		})
	}
}

2. 超时控制

func main() {
	p, _ := ants.NewPool(10)
	defer p.Release()

	// 带超时的提交
	err := p.Submit(func() {
		time.Sleep(2 * time.Second)
		fmt.Println("Task completed")
	})
	if err != nil {
		fmt.Println("Submit task error:", err)
	}
}

3. 动态调整池大小

func main() {
	p, _ := ants.NewPool(10)
	defer p.Release()

	// 调整容量到20
	p.Tune(20)

	// 调整容量到5
	p.Tune(5)
}

性能优化建议

1. 合理设置池大小：根据任务类型和服务器资源设置合适的池大小

   • CPU密集型任务：核心数+1
   • IO密集型任务：可以设置更大一些

2. 复用池：避免频繁创建和销毁池

3. 错误处理：注意处理池满时的错误

4. 监控指标：定期检查Running()和Cap()以调整池大小

完整示例

package main

import (
	"fmt"
	"sync"
	"time"

	"github.com/panjf2000/ants/v2"
)

func main() {
	// 创建容量为100的池，预分配内存
	p, _ := ants.NewPool(100, ants.WithPreAlloc(true))
	defer p.Release()

	var wg sync.WaitGroup
	task := func(i int) func() {
		return func() {
			time.Sleep(100 * time.Millisecond)
			fmt.Printf("Task %d done\n", i)
			wg.Done()
		}
	}

	start := time.Now()
	for i := 0; i < 1000; i++ {
		wg.Add(1)
		p.Submit(task(i))
	}

	wg.Wait()
	fmt.Printf("All tasks done, time elapsed: %s\n", time.Since(start))
	fmt.Printf("Pool size: %d, Running workers: %d\n", p.Cap(), p.Running())
}

ants库通过复用goroutine显著降低了频繁创建和销毁goroutine的开销，特别适合高并发场景。合理使用可以显著提升程序性能并降低资源消耗。