golang协程数量控制

最新推荐文章于 2024-08-08 14:28:32 发布

Madison001

最新推荐文章于 2024-08-08 14:28:32 发布

阅读量159

点赞数

CC 4.0 BY-SA版权

文章标签： golang 开发语言后端

本文链接：https://blog.youkuaiyun.com/Madison001/article/details/133040330

本文介绍了如何在Golang中通过创建协程池来限制并发的Goroutine数量，以避免资源耗尽和高CPU使用率。示例代码展示了一个简单的协程池实现，使用通道和 sync.WaitGroup 来控制协程的增减和等待执行完成。通过调用 Pool 的 Add 和 Done 方法，可以在执行 go 之前确保不超过预设的协程池大小。

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

协程数量控制

在Golang中，Goroutine虽然很好，但是数量太多了，往往会带来很多麻烦，比如耗尽系统资源导致程序崩溃，或者CPU使用率过高导致系统忙不过来。所以我们可以限制下Goroutine的数量,这样就需要在每一次执行go之前判断goroutine的数量，如果数量超了，就要阻塞go的执行。第一时间想到的就是使用通道。每次执行的go之前向通道写入值，直到通道满的时候就阻塞了.

package main

import (
	"fmt"
	"runtime"
	"sync"
	"time"
)
// Pool Goroutine Pool
type Pool struct {
	queue chan int
	wg *sync.WaitGroup
}
// New 新建一个协程池
func NewPool(size int) *Pool{
	if size <=0{
		size = 1
	}
	return &Pool{
		queue:make(chan int,size),
		wg:&sync.WaitGroup{},
	}
}
// Add 新增一个执行
func (p *Pool)Add(delta int){
	// delta为正数就添加
	for i :=0;i<delta;i++{
		p.queue <-1
	}
	// delta为负数就减少
	for i:=0;i>delta;i--{
		<-p.queue
	}
	p.wg.Add(delta)
}
// Done 执行完成减一
func (p *Pool) Done(){
	<-p.queue
	p.wg.Done()
}
// Wait 等待Goroutine执行完毕
func (p *Pool) Wait(){
	p.wg.Wait()
}

func main(){
	// 这里限制5个并发
	pool := NewPool(5)
	fmt.Println("the NumGoroutine begin is:",runtime.NumGoroutine())
	for i:=0;i<20;i++{
		pool.Add(1)
		go func(i int) {
			time.Sleep(time.Second)
			fmt.Println("the NumGoroutine continue is:",runtime.NumGoroutine())
			pool.Done()
		}(i)
	}
	pool.Wait()
	fmt.Println("the NumGoroutine done is:",runtime.NumGoroutine())
}