How can I prevent deadlock when using WaitGroups and buffered channels in Go?

Deadlock in Go: WaitGroup and Buffered Channels

In Go, deadlock occurs when concurrent goroutines wait indefinitely for each other to complete. One common cause of deadlock involves the use of WaitGroups and buffered channels.

Example of a Deadlock

Consider the following code:

<code class="go">package main

import "fmt"
import "sync"

func main() {
    ch := make(chan []int, 4)
    var m []int

    var wg sync.WaitGroup
    for i := 0; i <p>This code intends to send 5 empty slices to a buffered channel with a capacity of 4 and then read from the channel after all goroutines complete. However, the code results in a deadlock error.</p>
<p><strong>Cause of the Deadlock</strong></p>
<p>The deadlock arises due to two issues:</p>
<ol>
<li>
<strong>Insufficient Channel Buffer:</strong> The channel has a capacity of 4, which is too small for the 5 goroutines trying to send data. When the channel becomes full, subsequent goroutines waiting to send data (line 15) will block indefinitely.</li>
<li>
<strong>Blocking Channel Iteration:</strong> The loop that iterates over the channel (lines 22-24) blocks indefinitely because it waits for more elements to arrive on the channel. Since all goroutines have finished sending data and no more data is expected, this iteration will never complete without a corresponding goroutine reading from the channel.</li>
</ol>
<p><strong>Solution</strong></p>
<p>To resolve the deadlock, make one of the following modifications:</p>
<p><strong>Solution 1:</strong></p>
<p>Increase the channel capacity to 5 (or more) and close it after sending all data:</p>
<pre class="brush:php;toolbar:false"><code class="go">ch := make(chan []int, 5)
...
wg.Wait()
close(ch)</code>

Solution 2:

Start a separate goroutine to read from the channel and notify the main goroutine when all data has been received:

<code class="go">func main() {
    ch := make(chan []int, 4)
    var m []int

    var wg sync.WaitGroup
    for i := 0; i </code>

The above is the detailed content of How can I prevent deadlock when using WaitGroups and buffered channels in Go?. For more information, please follow other related articles on the PHP Chinese website!