Go 实现不同goroutine之间的阻塞

最新推荐文章于 2025-09-21 08:28:10 发布

原创最新推荐文章于 2025-09-21 08:28:10 发布 · 540 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#golang

GO 专栏收录该内容

2 篇文章

订阅专栏

本文探讨了Go语言中如何通过WaitGroup在main包的main()函数之外管理goroutine的生命周期，重点讲解了未使用WaitGroup导致的问题和加入WaitGroup后的执行流程，通过实例展示了WaitGroup在阻塞和信号传递中的关键作用。

Go 程序从 main 包的 main() 函数开始，在程序启动时，Go 程序就会为 main() 函数创建一个默认的 goroutine。

所有 goroutine 在 main() 函数结束时会一同结束。
若在启用的goroutine中不使用WaitGroup的话会因为main函数已执行完，阻塞的函数与发送信号的函数会一同结束，不能真正实现阻塞的功能。

因此可以使用WaitGroup来实现阻塞的功能。

如下为不加WaitGroup时的版本

package main

import (
	"fmt"
	"time"
)

var closeCh = make(chan struct{})

func main() {
	// var wg sync.WaitGroup
	fmt.Println("start main func")

	// wg.Add(1)
	go func() {
		fmt.Println("waiting for signal")
		<-closeCh
		fmt.Println("got signal.")
		// wg.Done()
	}()

	// wg.Add(1)
	go func() {
		fmt.Println("preparing for signal:")
		for i := 0; i < 3; i++ {
			fmt.Println(">>>>")
			time.Sleep(time.Second * 1)
		}
		closeCh <- struct{}{}
		fmt.Println("sent signal.")
		// wg.Done()
	}()
	// wg.Wait()
}

执行结果

start main func // 只打印输出了这么一行

如下为加WaitGroup时的版本

package main

import (
	"fmt"
	"sync"
	"time"
)

var closeCh = make(chan struct{})

func main() {
	var wg sync.WaitGroup
	fmt.Println("start main func")

	wg.Add(1)
	go func() {
		fmt.Println("waiting for signal")
		<-closeCh
		fmt.Println("got signal.")
		wg.Done()
	}()

	wg.Add(1)
	go func() {
		fmt.Println("preparing for signal:")
		for i := 0; i < 3; i++ {
			fmt.Println(">>>>")
			time.Sleep(time.Second * 1)
		}
		closeCh <- struct{}{}
		fmt.Println("sent signal.")
		wg.Done()
	}()
	wg.Wait()
}

执行结果：

start main func
preparing for signal:
>>>>
waiting for signal
>>>>
>>>>
got signal.
sent signal.

参考：
http://c.biancheng.net/view/93.html