golang協程數量控制

協程數量控制

在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())
}