Go语言中的同步等待组WaitGroup

    在Go语言的sync包中,提供了基本的同步单元,如互斥锁。除了Once和WaitGroup类型,⼤部分都是适⽤于低⽔平程序线程,高水平的同步使⽤channel通信更好⼀些。
    我们可以通过WaitGroup里的wait()方法,来阻塞主线程,直到其他子线程或协程运行完毕,再回到主线程。

WaitGroup同步等待组

type WaitGroup struct {
    noCopy noCopy
    state1 [12]byte
    sema uint32
}

    WaitGroup⽤于等待⼀组线程或协程的结束。⽗线程调⽤Add⽅法来设定应等待的线程数量。每个被等待的线程在结束时应调⽤Done⽅法。同时,主线程⾥可以调⽤Wait⽅法阻塞自身,等待Add()方法里声明的线程运行至结束,再返回主线程。

WaitGroup里的方法

  • func (wg *WaitGroup) Add(delta int)
    Add⽅法向内部计数加上delta, delta可以是负数;如果内部计数器变为0,Wait⽅法阻塞等待的所有线程都会释放;如果计数器⼩于0,⽅法panic。注意,Add加上正数的调⽤应在Wait之前,否则Wait可能只会等待很少的线程。⼀般来说,Add()⽅法应在创建新的线程,或者其他应等待的事件之前调用。
  • func (wg *WaitGroup) Done()
    Done⽅法减少WaitGroup计数器的值,应在线程的最后执行。
  • func (wg *WaitGroup) Wait()
    Wait⽅法阻塞直到WaitGroup计数器减为0。

    案例1. WaitGroup通过wait()阻塞主函数,来运行3个子协程。
//myWaitGroupDes.go

// myWatiGroupDes project main.go
package main

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

func printNumA(wg *sync.WaitGroup) {
	k := 0
	for i := 11; i <= 20; i += 2 {
		fmt.Printf("子协程A[%d]= %d \n", k, i)
		time.Sleep(time.Duration(rand.Intn(1000)))
		k++
	}
	wg.Done() //计算器减1
}

func printNumB(wg *sync.WaitGroup) {
	k := 0
	for i := 22; i <= 30; i += 2 {
		fmt.Printf("子协程B[%d]= %d \n", k, i)
		time.Sleep(time.Duration(rand.Intn(1000)))
		k++
	}
	wg.Done() //计算器减1
}

func printNumC(wg *sync.WaitGroup) {
	k := 0
	for i := 'a'; i <= 'e'; i++ {
		fmt.Printf("子协程C[%d]= %c \n", k, i)
		time.Sleep(time.Duration(rand.Intn(1000)))
		k++
	}
	wg.Done() //计算器减1
}

func main() {
	var wg sync.WaitGroup
	fmt.Printf("%T\n", wg)
	fmt.Println(wg)
	wg.Add(3)
	rand.Seed(time.Now().UnixNano())

	go printNumA(&wg)
	go printNumB(&wg)
	go printNumC(&wg)

	wg.Wait()
	fmt.Println("main解除阻塞,main is over...")
}


    效果如下:

图(1) 阻塞主线程,来运行3个子协程
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Go语言的channel、sync.WaitGroup和context是三个非常重要的并发编程工具。下面我将对它们进行详细介绍。 ### Channel Go语言的channel是一种在多个goroutine之间进行通信的机制。也可以说,channel是一种数据结构,它可以让一个goroutine向另一个goroutine发送一个值,同时还可以让另一个goroutine从channel接收这个值。在Go语言,使用make函数创建一个channel。例如: ```go ch := make(chan int) ``` 这行代码创建了一个类型为int的channel。可以在goroutine使用ch <- value语句向channel发送一个整数,例如: ```go go func() { ch <- 1 }() ``` 可以使用value := <- ch语句从channel接收一个整数,例如: ```go value := <- ch ``` 这行代码会阻塞,直到有一个整数被发送到这个channel为止。需要注意的是,如果没有接收者,发送操作会一直阻塞,直到有接收者为止;如果没有发送者,接收操作也会一直阻塞,直到有发送者为止。 ### sync.WaitGroup sync.WaitGroupGo语言的一个同步工具,它可以等待goroutine完成工作。在WaitGroup,每个goroutine的工作完成后,都需要调用Done方法。主goroutine可以在Wait方法上阻塞,等待所有的goroutine完成工作。例如: ```go var wg sync.WaitGroup for i := 0; i < 10; i++ { wg.Add(1) go func() { // do some work wg.Done() }() } wg.Wait() ``` 这行代码创建了一个WaitGroup,并且启动了10个goroutine进行工作。每个goroutine完成工作后,都会调用wg.Done方法,主goroutine在wg.Wait上阻塞,等待所有的goroutine完成工作。 ### context context是Go语言的一个用于传递请求范围数据的机制。在一个请求处理,可以使用context携带一些请求数据,同时也可以使用context取消请求处理。例如: ```go func handleRequest(ctx context.Context) { // do some work select { case <-ctx.Done(): // handle cancelation default: // continue working } } ``` 这行代码定义了一个处理请求的函数,该函数接收一个context参数。如果context被取消,处理请求的函数将会停止工作。例如: ```go ctx, cancel := context.WithCancel(context.Background()) go func() { time.Sleep(time.Second) cancel() }() handleRequest(ctx) ``` 这行代码创建了一个带有取消功能的context,并且启动了一个goroutine在1秒后取消context。handleRequest函数会使用这个context来处理请求,并且如果context被取消,handleRequest函数会立刻停止工作。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值