Go语言启动4个协程处理10个任务的例子

Go语言启动4个协程处理10个任务的例子

package main

import (
    "fmt"
    "math/rand"
    "sync"
    "time"
)

const (
    numberGoroutines = 4
    taskLoad         = 10
)

var wg sync.WaitGroup

func init() {
    rand.Seed(time.Now().Unix())
}

func main() {
    tasks := make(chan string, taskLoad)
    wg.Add(numberGoroutines)
    for i := 1; i <= numberGoroutines; i++ {
        go work(tasks, i)
    }

    for post := 1; post <= taskLoad; post++ {
        tasks <- fmt.Sprintf("Task:%d", post)
    }
    close(tasks)
    wg.Wait()
}

func work(tasks chan string, worker int) {
    defer wg.Done()
    for {
        task, ok := <-tasks
        if !ok {
            fmt.Printf("worker:%d shutting down(关闭)\n", worker)
            return
        }
        fmt.Printf("Worker:%d started %s\n", worker, task)
        sleep := rand.Int63n(100)
        time.Sleep(time.Duration(sleep) * time.Millisecond)
        fmt.Printf("Worker:%d completed %s\n", worker, task)
    }
}

输出：

Worker:4 started Task:1
Worker:2 started Task:2
Worker:1 started Task:4
Worker:3 started Task:3
Worker:4 completed Task:1
Worker:4 started Task:5
Worker:3 completed Task:3
Worker:3 started Task:6
Worker:3 completed Task:6
Worker:3 started Task:7
Worker:1 completed Task:4
Worker:1 started Task:8
Worker:2 completed Task:2
Worker:2 started Task:9
Worker:2 completed Task:9
Worker:2 started Task:10
Worker:3 completed Task:7
worker:3 shutting down(关闭)
Worker:4 completed Task:5
worker:4 shutting down(关闭)
Worker:1 completed Task:8
worker:1 shutting down(关闭)
Worker:2 completed Task:10
worker:2 shutting down(关闭)