golang協程池

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

const Max = 10000000type GoroutinePoll struct {    Queue          chan func() error    Total, Num     int    Result         chan error    FinishCallBack func() error}func (g *GoroutinePoll) Init(total int) {    g.Total = total    g.Queue = make(chan func() error, total)    g.Result = make(chan error, total)}func (g *GoroutinePoll) SetFinishCallBack(f func() error) {    g.FinishCallBack = f}func (g *GoroutinePoll) Add(f func() error) error {    if len(g.Queue) == g.Total || g.Total == 0 {        return errors.New(" add goroutine fail!")    }    g.Queue <- f    g.Num++    return nil}func (g *GoroutinePoll) Start() {    for i := 0; i < g.Num; i++ {        v, ok := <-g.Queue        if !ok {            break        }        go func() {            err := v()            g.Result <- err        }()    }    for i := 0; i < g.Num; i++ {        err, ok := <-g.Result        if !ok {            break        }        if err != nil {            fmt.Println(err)        }    }    if err := g.FinishCallBack(); err != nil {        fmt.Println("執行回呼函數失敗！", err)    }}func (g *GoroutinePoll) Close() {    close(g.Queue)    close(g.Result)}func main() {    runtime.GOMAXPROCS(16)    total := 10    g := GoroutinePoll{}    g.Init(total)    finshCallBack := func() error {        fmt.Println("callback !")        return nil    }    g.SetFinishCallBack(finshCallBack)    f1 := func() error {        for i := 0; i < Max; i++ {        }        return nil    }    g.Add(f1)    f2 := func() error {        for i := 0; i < Max; i++ {        }        return nil    }    g.Add(f2)    f3 := func() error {        for i := 0; i < Max; i++ {        }        return nil    }    g.Add(f3)    f4 := func() error {        for i := 0; i < Max; i++ {        }        return nil    }    g.Add(f4)    now := time.Now()    g.Start()    g.Close()    fmt.Println(time.Since(now))    now = time.Now()    f1()    f2()    f3()    f4()    fmt.Println(time.Since(now))}

callback !
3.863554ms
12.890933ms