快捷導(dǎo)航

grpool?goroutine池協(xié)程管理

更新時(shí)間：2022年06月08日 10:38:39 作者：??王中陽Go????

這篇文章主要介紹了grpool?goroutine池協(xié)程管理，goroutine協(xié)程非常輕量級(jí)，這也是為什么go支持高并發(fā)，但是goroutine頻繁創(chuàng)建銷毀對(duì)GC的壓力比較大，文章圍繞主題展開詳細(xì)的內(nèi)容介紹，需要的小伙伴可以參考一下

前言

goroutine協(xié)程非常輕量級(jí)，這也是為什么go支持高并發(fā)，但是goroutine頻繁創(chuàng)建銷毀對(duì)GC的壓力比較大。

grpool的作用就是復(fù)用goroutine，減少頻繁創(chuàng)建銷毀的性能消耗。

名詞概念

Pool: goroutine池，用于管理若干可復(fù)用的goroutine協(xié)程資源

Worker: 池對(duì)象中參與任務(wù)執(zhí)行的goroutine，一個(gè)worker可以執(zhí)行若干個(gè)job，直到隊(duì)列中再無等待的job

Job：添加到池對(duì)象的任務(wù)隊(duì)列中等待執(zhí)行的任務(wù)，是一個(gè)func()方法，一個(gè)job同時(shí)只能被一個(gè)worker獲取并執(zhí)行。

使用示例

使用默認(rèn)的協(xié)程池，限制100個(gè)協(xié)程執(zhí)行1000個(gè)任務(wù)

pool.Size() 獲得當(dāng)前工作的協(xié)程數(shù)量

pool.Jobs() 獲得當(dāng)前池中待處理的任務(wù)數(shù)量

package main
import (
   "fmt"
   "github.com/gogf/gf/os/grpool"
   "github.com/gogf/gf/os/gtimer"
   "sync"
   "time"
)
func main() {
   pool := grpool.New(100)
   //添加1千個(gè)任務(wù)
   for i := 0; i < 1000; i++ {
      _ = pool.Add(job)
   }
   fmt.Println("worker:", pool.Size()) //當(dāng)前工作的協(xié)程數(shù)量
   fmt.Println("jobs:", pool.Jobs())   //當(dāng)前池中待處理的任務(wù)數(shù)量

   gtimer.SetInterval(time.Second, func() {
      fmt.Println("worker:", pool.Size()) //當(dāng)前工作的協(xié)程數(shù)
      fmt.Println("jobs:", pool.Jobs())   //當(dāng)前池中待處理的任務(wù)數(shù)
   })
   //阻止進(jìn)程結(jié)束
   select {}
}
//任務(wù)方法
func job() {
   time.Sleep(time.Second)
}

打印結(jié)果：

是不是灰常簡單~

踩坑之旅

一個(gè)簡單的場(chǎng)景，請(qǐng)使用協(xié)程打印0~9。

常犯的錯(cuò)誤

大家看下面的代碼有沒有問題，請(qǐng)預(yù)測(cè)一下打印結(jié)果。

wg := sync.WaitGroup{}
for i := 0; i < 9; i++ {
   wg.Add(1)
   go func() {
      fmt.Println(i)
      wg.Done()
   }()
}
wg.Wait()

不用著急看答案

猜一下打印結(jié)果是什么

打印結(jié)果：

分析原因

對(duì)于異步線程/協(xié)程來講，函數(shù)進(jìn)行異步執(zhí)行注冊(cè)時(shí)，該函數(shù)并未真正開始執(zhí)行(注冊(cè)時(shí)只在goroutine的棧中保存了變量i的內(nèi)存地址)，而一旦開始執(zhí)行時(shí)函數(shù)才會(huì)去讀取變量i的值，而這個(gè)時(shí)候變量i的值已經(jīng)自增到了9。

正確寫法：

wg := sync.WaitGroup{}
for i := 0; i < 9; i++ {
   wg.Add(1)
   go func(v int) {
      fmt.Println(v)
      wg.Done()
   }(i)
}
wg.Wait()

打印結(jié)果：

使用grpool

使用grpool和使用go一樣，都需要把當(dāng)前變量i的值賦值給一個(gè)不會(huì)改變的臨時(shí)變量，在函數(shù)中使用該臨時(shí)變量而不是直接使用變量i。

錯(cuò)誤代碼

wg := sync.WaitGroup{}
for i := 0; i < 9; i++ {
   wg.Add(1)
   _ = grpool.Add(func() {
      fmt.Println(i) //打印結(jié)果都是9
      wg.Done()
   })
}
wg.Wait()

打印結(jié)果：

正確代碼

wg := sync.WaitGroup{}
for i := 0; i < 9; i++ {
   wg.Add(1)
   v := i //grpoll.add() 的參數(shù)只能是不帶參數(shù)的匿名函數(shù) 因此只能以設(shè)置臨時(shí)變量的方式賦值
   _ = grpool.Add(func() {
      fmt.Println(v)
      wg.Done()
   })
}
wg.Wait()

打印結(jié)果：