Go并发死锁解决技巧：Goexit与WaitGroup实战指南

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《Go并发死锁解决：Goexit与WaitGroup实战技巧》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

本文旨在帮助Go语言初学者理解和解决并发代码中常见的deadlock问题。通过分析一个包含runtime.Goexit()和time.After()的示例代码，我们将深入探讨Goexit()的正确使用方式以及如何利用sync.WaitGroup来优雅地等待goroutine完成，从而避免deadlock的发生。同时，本文也解释了在Go Playground环境下使用time.After()可能出现的特殊情况。

理解runtime.Goexit()

runtime.Goexit()函数用于终止调用它的goroutine。根据Go官方文档的描述，Goexit()只会终止当前goroutine，而不会影响其他goroutine的执行。在goroutine终止之前，所有通过defer语句注册的函数都会被执行。

在示例代码中，defer runtime.Goexit()被放置在main函数的开头。这意味着当main函数执行完毕时，它会尝试调用Goexit()来终止main goroutine。然而，由于main函数启动了多个子goroutine（通过go check(i)），并且这些子goroutine可能仍在运行，此时终止main goroutine会导致程序无法正确等待所有子goroutine完成，从而引发deadlock。更具体地说，当所有子goroutine完成后，没有其他goroutine在运行，Go运行时会检测到这种情况，并抛出 "all goroutines are asleep - deadlock!" 错误。

因此，在main函数中使用runtime.Goexit()通常是不正确的，除非你有明确的需求要提前终止程序。

使用sync.WaitGroup等待Goroutine完成

为了避免deadlock并确保所有goroutine都执行完毕，可以使用sync.WaitGroup。sync.WaitGroup提供了一种机制来等待一组goroutine完成。

以下是使用sync.WaitGroup的修改后的代码：

package main

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

func check(id int, wg *sync.WaitGroup) {
    defer wg.Done() // 在goroutine结束时调用Done()
    fmt.Println("Checked", id)
    <-time.After(time.Duration(id) * time.Millisecond)
    fmt.Println("Woke up", id)
}

func main() {
    var wg sync.WaitGroup
    for i := 0; i <= 10; i++ {
        wg.Add(1) // 每次启动一个goroutine，计数器加1
        fmt.Println("Called with", i)
        go check(i, &wg)
    }
    wg.Wait() // 等待所有goroutine完成
    fmt.Println("Done for")
}

代码解释：

1. var wg sync.WaitGroup: 声明一个sync.WaitGroup类型的变量wg。
2. wg.Add(1): 在每次启动一个新的goroutine之前，调用wg.Add(1)将计数器加1。这表示有一个新的goroutine需要等待。
3. defer wg.Done(): 在check函数的开头使用defer wg.Done()。Done()方法会将计数器减1。defer关键字确保在check函数结束时，无论是否发生错误，Done()都会被调用。
4. wg.Wait(): 在main函数的最后，调用wg.Wait()。Wait()方法会阻塞当前goroutine（即main goroutine），直到计数器变为0，表示所有通过Add()添加的goroutine都已经调用了Done()方法。

通过使用sync.WaitGroup，main函数会等待所有子goroutine执行完毕后才继续执行，从而避免了deadlock的发生。

Go Playground中的time.After()问题

需要注意的是，在Go Playground环境中，time.After()可能会导致deadlock。这是因为Go Playground对时间进行了冻结，使得time.After()永远不会返回。因此，在Go Playground中测试涉及时间相关的代码时，需要特别注意。

总结：

避免Go并发代码中的deadlock需要仔细考虑goroutine的生命周期和同步问题。不要在main函数中随意使用runtime.Goexit()，而是应该使用sync.WaitGroup来确保所有goroutine都执行完毕。同时，在Go Playground中测试时间相关的代码时，需要注意其特殊性。理解这些概念可以帮助你编写更健壮和可靠的Go并发程序。

以上就是《Go并发死锁解决技巧：Goexit与WaitGroup实战指南》的详细内容，更多关于的资料请关注golang学习网公众号！