Go语言中goroutine和WaitGroup的使用示例详解

Go语言中goroutine和WaitGroup的使用

介绍

goroutine 是Go中一个轻量级的线程, 只需要一个go关键字就可以创建一个goroutine

WaitGroup则是用于多个goroutine 协作的时候使用, 可以指定多个goroutine 一起完成指定动作之后执行对应的逻辑。

例如:100M赛跑,要等所有的运动员都就位,才能发令开始, 这个类似于Java中的countDownlatch

例子

package main

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

func main() {
        // 随机数种子
        rand.Seed(time.Now().Unix())
        // 生命WaitGroup, 指定五个
        var wait sync.WaitGroup
        wait.Add(5)
        for i := 0; i < 5; i++ {
                i := i
                go func() {
                        waitTime := rand.Intn(10)
                        time.Sleep(time.Duration(waitTime) * time.Second)
                        fmt.Println("I'm " + strconv.Itoa(i) + ". sleep" + strconv.Itoa(waitTime))
                        // 表示准备好了
                        wait.Done()
                }()
        }
        fmt.Println("Wait  start")
        // 等待所有人准备好
        wait.Wait()
        fmt.Println("All  start")
}

以上程序输出结果

Wait start

I'm 3. sleep1

I'm 1. sleep2

I'm 2. sleep3

I'm 4. sleep4

I'm 0. sleep8

All start

会在wait.Wait()处卡住,等待执行了五次wait.Done() ,才放行。

扩展:Go的goroutine与sync.WaitGroup的初步使用

goroutine的使用以及sync.WaitGroup

emmm,没啥好介绍的,就简单的使用。

package main

import (
    "fmt"
    "sync"
)

//协程计数器
var wg sync.WaitGroup

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

    for i := 0; i < 5; i++ {
        wg.Add(1)//协程计数器加1
        go func(x int) {
            defer wg.Done()// 协程计数器减1
            fmt.Println(x)
        }(i)
    }

    wg.Wait()// 等待所有的协程执行完毕
    fmt.Println("主线程退出")
}

原文地址:https://blog.csdn.net/qq_21047625/article/details/124530785