码迷,mamicode.com
首页 > 编程语言 > 详细

Golang 线程池

时间:2018-10-31 17:51:49      阅读:273      评论:0      收藏:0      [点我收藏+]

标签:span   package   one   []   style   pool   fun   print   close   

经常会用到协程,但是不能一下开那么多协调,只需要 poolSize 个即可,多了不行。这些个协程在执行完后必须等其完成之后才能进行下一步动作。假定工作方法为 work 。

 

package main

import (
    "fmt"
    "runtime"
    "sync"
    "time"
)
// WAIT GROUP
var wg sync.WaitGroup
var IDS_ALL = []int {40,42,43,44,164,166,171,173,174,175,177,518,192,193}

func main () {
    fmt.Println("BEGIN")

    poolSize := runtime.NumCPU()
    runtime.GOMAXPROCS(poolSize)

    ch := make(chan int,poolSize)
    for _,catid := range IDS_ALL {
        ch <- 1
        wg.Add(1)
        go work(catid,ch)
    }
    wg.Wait()
    close(ch)

    fmt.Println("DONE")
}

// 工作
func work(catid int,ch chan int) {

    fmt.Println("WORKING ",catid)
    time.Sleep(2 * 1e9)

    wg.Done()
    <-ch
}

 

 

就是这样。有问题,但能用。

Golang 线程池

标签:span   package   one   []   style   pool   fun   print   close   

原文地址:https://www.cnblogs.com/qufo/p/9884478.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!