golang工作者goroutine池模型

這是一個建立於 的文章，其中的資訊可能已經有所發展或是發生改變。

在golang編寫服務程式過程中，如果每次來鬥啟動一個goroutine去處理任務，處理一個任務後就退出，這樣勢必會造成資源浪費。構建一個工作goroutine池來處理任務相對資源利用會少些，具體的情況需對比測試。

首先定義工作goroutine池，內部定義了兩個變數，一個是任務隊列，一個是需啟動goroutine的數量

type WorkerPool struct {    tasks    <-chan *string //任務隊列長度    poolSize int            //啟動goroutine的數目}

實現工作函數

func (p *WorkerPool) Run() {    var wg sync.WaitGroup    for i := 0; i < p.poolSize; i++ {        wg.Add(1)        go func() {            for task := range p.tasks {                fmt.Println("Consume Task", *task)            }            wg.Done()        }()    }    wg.Wait()    fmt.Println("WorkerPool | Pool exit.")}

以下是測試代碼

func main() {    taskNum := 100    projects := make(chan *string, taskNum)    //啟動生產任務goroutine    go func() {        for i := 0; ; i++ {            s := "Project " + fmt.Sprintf("%d", i)            fmt.Println("Produce task", s)            projects <- &s            time.Sleep(5 * time.Millisecond)        }    }()    p := WorkerPool{        tasks:    projects,        poolSize: 10,    }    p.Run()}