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

时间:2023-03-23 水淹萌龙人气:0

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("主线程退出")
}

加载全部内容