Go语言中控制线程的信号量WaitGroup

demo程序如下,有注释

package main

import (
	"fmt"
	"sync"
)

func main() {
	//声明并初始化一个数组,里面存放的是线程id
	threadIds := []int{1, 2, 4, 6, 8}
	//声明一个WaitGroup类型变量,WaitGroup类似Java中的信号量java.util.concurrent.Semaphore
	var waitGroup sync.WaitGroup
	//设置等待的线程数量
	waitGroup.Add(len(threadIds))

	//_是数组下标,ele是数组元素
	for _, ele := range threadIds {
		//声明一个匿名函数,并用go命令启动一个线程(goroutine)
		go func( ele int) {
			//在每个线程中输出线程id号对应的数量个数字
			for i := 0; i < ele; i++ {
				fmt.Printf("线程%d 输出%d \n", ele, i)
			}
			//每个线程结束后,就调用Done,执行线程数量减一操作
			waitGroup.Done()
		}(ele)

	}

	//主线程阻塞住,等待信号量中的线程执行完
	waitGroup.Wait()
	fmt.Println("wait结束")

	fmt.Println("主函数结束")
}

输出结果如下(由于线程顺序的随机性,每次执行输出的线程顺序可能不一样):

线程8 输出0
线程8 输出1
线程8 输出2
线程8 输出3
线程8 输出4
线程8 输出5
线程8 输出6
线程8 输出7
线程1 输出0
线程4 输出0
线程4 输出1
线程4 输出2
线程4 输出3
线程2 输出0
线程2 输出1
线程6 输出0
线程6 输出1
线程6 输出2
线程6 输出3
线程6 输出4
线程6 输出5
wait结束
主函数结束

Process finished with exit code 0

猜你喜欢

转载自blog.csdn.net/lzufeng/article/details/86590079