How Can I Gracefully Exit a Go Program After All Goroutines Have Finished?

Ensuring Graceful Program Exit by Properly Waiting for Go Routines

When working with Go routines, it's crucial to ensure they complete before exiting your program. By understanding the mechanism behind waiting for routines to finish, you can implement it effectively.

Unveiling the Mechanism of "<- done"

The key to waiting for a Go routine is using the "<- done" statement, which is a blocking operation. This means that your program will pause until a value is sent to the channel. In the example you provided, the "done" channel is used to signal the completion of the "do_stuff" routine. Once "done <- true" is executed, the "<- done" statement will unblock, allowing your program to continue.

Avoiding Deadlocks Through Proper Synchronization

When you uncomment the last line, you encounter a deadlock because the channel "done" has already been read once. Deadlocks occur when two or more routines attempt to acquire resources that are held by each other, creating a dependency that prevents them from progressing. To avoid this, ensure proper synchronization between routines and channels.

In scenarios where you want to parallelize long-running functions, the "sync" package provides a more robust solution. The "sync.WaitGroup" type allows you to track the number of active routines and wait for them to complete before proceeding. The example below demonstrates this approach:

package main

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

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func() {
            longOp()
            wg.Done()
        }()
    }

    // Wait for all routines to finish
    wg.Wait()
}

func longOp() {
    time.Sleep(time.Second * 2)
    fmt.Println("Long operation completed")
}

By using the "sync.WaitGroup," you ensure that your program waits for all long-running functions to complete before proceeding, avoiding potential deadlocks. This provides a more controlled approach to coordinating Go routines.

The above is the detailed content of How Can I Gracefully Exit a Go Program After All Goroutines Have Finished?. For more information, please follow other related articles on the PHP Chinese website!