1.31 Golang工作池示例

在这个例子中,我们来看一下如何使用gorouotine和channel来实现工作池。

package mainimport "fmt"import "time"// 我们将在worker函数里面运行几个并行实例,这个函数从jobs通道// 里面接受任务,然后把运行结果发送到results通道。每个job我们// 都休眠一会儿,来模拟一个耗时任务。func worker(id int, jobs <-chan int, results chan<- int) {    for j := range jobs {        fmt.Println("worker", id, "processing job", j)        time.Sleep(time.Second)        results <- j * 2    }}func main() {    // 为了使用我们的工作池,我们需要发送工作和接受工作的结果,    // 这里我们定义两个通道,一个jobs,一个results    jobs := make(chan int, 100)    results := make(chan int, 100)    // 这里启动3个worker协程,一开始的时候worker阻塞执行,因为    // jobs通道里面还没有工作任务    for w := 1; w <= 3; w++ {        go worker(w, jobs, results)    }    // 这里我们发送9个任务,然后关闭通道,告知任务发送完成    for j := 1; j <= 9; j++ {        jobs <- j    }    close(jobs)    // 然后我们从results里面获得结果    for a := 1; a <= 9; a++ {        <-results    }

运行结果

worker 1 processing job 1worker 2 processing job 2worker 3 processing job 3worker 1 processing job 4worker 3 processing job 5worker 2 processing job 6worker 1 processing job 7worker 3 processing job 8worker 2 processing job 9

Leave a Reply

Your email address will not be published. Required fields are marked *