這篇文章主要為大家展示了“如何通過Channel實(shí)現(xiàn)Goroutine Pool”，內(nèi)容簡(jiǎn)而易懂，條理清晰，希望能夠幫助大家解決疑惑，下面讓小編帶領(lǐng)大家一起研究并學(xué)習(xí)一下“如何通過Channel實(shí)現(xiàn)Goroutine Pool”這篇文章吧。

創(chuàng)新互聯(lián)建站作為成都網(wǎng)站建設(shè)公司，專注網(wǎng)站建設(shè)、網(wǎng)站設(shè)計(jì)，有關(guān)企業(yè)網(wǎng)站設(shè)計(jì)方案、改版、費(fèi)用等問題，行業(yè)涉及成都衛(wèi)生間隔斷等多個(gè)領(lǐng)域，已為上千家企業(yè)服務(wù)，得到了客戶的尊重與認(rèn)可。

最近用到了 Go 從 Excel 導(dǎo)數(shù)據(jù)到服務(wù)器內(nèi)部 用的是 http 請(qǐng)求
但是發(fā)現(xiàn)一個(gè)問題 從文件讀取之后 新開 Goroutine 會(huì)無限制新增
導(dǎo)致全部卡在初始化請(qǐng)求 于是乎就卡死了

問題模擬

• 模擬代碼

func main() {
    pool := sync.WaitGroup{}
    for i := 0; i < 500; i++ {
        pool.Add(1)
        go func(i int) {
            resp, err := http.Get("http://ip.3322.org")
            if err != nil {
                fmt.Println(i, err)
            } else {
                defer resp.Body.Close()
                result, _ := ioutil.ReadAll(resp.Body)
                fmt.Println(i, string(result))
            }
            pool.Done()
        }(i)
    }
    pool.Wait()
}

• 數(shù)量小的情況下 沒有問題 但是數(shù)量比較大的情況 就會(huì)發(fā)現(xiàn)程序直接卡死 一段時(shí)間之后報(bào)錯(cuò) 并且沒有發(fā)出任何請(qǐng)求

問題解決

• 實(shí)際上看的出來 是應(yīng)為同時(shí)發(fā)起了太多的HTTP請(qǐng)求 導(dǎo)致系統(tǒng)卡死 數(shù)據(jù)沒有發(fā)送

• 想到我在Java中用Thread提交請(qǐng)求 我就考慮 可不可限制 Goroutine 的數(shù)量

• 使用強(qiáng)大的百度 果然找到了大佬已經(jīng)寫好的協(xié)程池

• 代碼如下 我加上了注釋

package gopool

import (
    "sync"
)

// Pool Goroutine Pool
type Pool struct {
    queue chan int
    wg    *sync.WaitGroup
}

// New 新建一個(gè)協(xié)程池
func New(size int) *Pool {
    if size <= 0 {
        size = 1
    }
    return &Pool{
        queue: make(chan int, size),
        wg:    &sync.WaitGroup{},
    }
}

// Add 新增一個(gè)執(zhí)行
func (p *Pool) Add(delta int) {
    // delta為正數(shù)就添加
    for i := 0; i < delta; i++ {
        p.queue <- 1
    }
    // delta為負(fù)數(shù)就減少
    for i := 0; i > delta; i-- {
        <-p.queue
    }
    p.wg.Add(delta)
}

// Done 執(zhí)行完成減一
func (p *Pool) Done() {
    <-p.queue
    p.wg.Done()
}

// Wait 等待Goroutine執(zhí)行完畢
func (p *Pool) Wait() {
    p.wg.Wait()
}

• 然后修改剛才的測(cè)試方法

package main

import (
    "io/ioutil"
    "log"
    "net/http"
    "yumc.pw/cloud/lib/gopool"
)

func main() {
    // 這里限制5個(gè)并發(fā)
    pool := gopool.New(5)// sync.WaitGroup{}
    for i := 0; i < 500; i++ {
        pool.Add(1)
        go func(i int) {
            resp, err := http.Get("http://ip.3322.org")
            if err != nil {
                fmt.Println(i, err)
            } else {
                defer resp.Body.Close()
                result, _ := ioutil.ReadAll(resp.Body)
                fmt.Println(i, string(result))
            }
            pool.Done()
        }(i)
    }
    pool.Wait()
}

以上是“如何通過Channel實(shí)現(xiàn)Goroutine Pool”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對(duì)大家有所幫助，如果還想學(xué)習(xí)更多知識(shí)，歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道！