waitgroup死锁通常由add()和done()调用不匹配或重复使用导致。要避免死锁,需确保三点:1. 启动goroutine前调用add(1),并在对应goroutine末尾使用defer wg.done()保证成对执行;2. 避免在wait()之后再次调用add(),应一次性使用waitgroup,重复需求需新建实例;3. add()参数不得使计数器为负,否则引发panic。此外,golang同步机制还包括mutex、rwmutex、channel、cond和atomic,适用于不同并发场景。

WaitGroup是Golang中用于等待一组goroutine完成的同步机制。你可以把它想象成一个计数器,主goroutine等待这个计数器归零。

WaitGroup的核心在于三个方法:
Add(delta int)、
Done()和
Wait()。
Add()增加计数器的值,
Done()减少计数器的值,
Wait()阻塞直到计数器归零。

package main
import (
"fmt"
"sync"
"time"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done() // 确保goroutine结束时计数器减一
fmt.Printf("Worker %d starting\n", id)
time.Sleep(time.Second) // 模拟耗时操作
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 3; i++ {
wg.Add(1) // 启动一个goroutine,计数器加一
go worker(i, &wg)
}
wg.Wait() // 等待所有goroutine完成
fmt.Println("All workers done!")
}如何避免WaitGroup使用中的常见死锁?
死锁往往发生在
Add()和
Done()调用不匹配的时候。例如,
Add()的调用次数少于实际启动的goroutine数量,或者
Done()被调用次数多于
Add()的调用次数。前者会导致
Wait()永久阻塞,后者会导致panic。
立即学习“go语言免费学习笔记(深入)”;
避免死锁的关键在于确保
Add()和
Done()的调用次数严格匹配。一种好的做法是在启动goroutine之前立即调用
Add(),并在goroutine结束时立即调用
Done()。可以使用
defer关键字来确保
Done()一定会被执行,即使goroutine发生panic。

另一种可能导致死锁的情况是在
Wait()之后再次调用
Add()。WaitGroup的设计初衷是一次性的,不应该重复使用。如果需要重复使用,应该创建一个新的WaitGroup实例。
WaitGroup的Add方法参数为负数会发生什么?
Add()方法接受一个整数参数,表示计数器需要增加的值。如果传递一个负数,实际上是减少计数器的值。如果计数器变为负数,会引发panic。
package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
wg.Add(1)
wg.Done() // 计数器变为0
wg.Add(-2) // 计数器变为-2,会panic
fmt.Println("This line will not be printed")
}因此,在使用
Add()方法时,务必确保传递的参数不会导致计数器变为负数。
除了WaitGroup,Golang还有哪些同步机制?
除了WaitGroup,Golang还提供了其他的同步机制,例如:
- Mutex (互斥锁): 用于保护共享资源,防止多个goroutine同时访问。
- RWMutex (读写锁): 允许多个goroutine同时读取共享资源,但只允许一个goroutine写入。
- Channel (通道): 用于goroutine之间的通信和同步。
- Cond (条件变量): 用于goroutine在满足特定条件时进行等待和唤醒。
- Atomic (原子操作): 用于对基本数据类型进行原子操作,避免数据竞争。
选择哪种同步机制取决于具体的应用场景。WaitGroup适用于等待一组goroutine完成,Mutex和RWMutex适用于保护共享资源,Channel适用于goroutine之间的通信,Cond适用于复杂的条件同步,Atomic适用于简单的原子操作。选择合适的同步机制可以提高程序的性能和可靠性。










