Comment attendre gracieusement la fin des routines Go ?

Comment attendre gracieusement la fin des routines Go

Dans Go, la concurrence est obtenue grâce aux goroutines, qui sont des threads légers qui s'exécutent simultanément avec la routine principale. Pour garantir que le programme ne se termine pas avant que toutes les goroutines n'aient terminé leurs tâches, il est crucial d'attendre correctement qu'elles se terminent.

Utiliser un canal pour signaler l'achèvement

Une approche courante pour attendre les goroutines consiste à utiliser un canal booléen. Lorsqu'une goroutine termine sa tâche, elle envoie une vraie valeur au canal, et la routine principale attend cette valeur en recevant du canal.

func do_stuff(done chan bool) {
    fmt.Println("Doing stuff")
    done <- true
}

func main() {
    fmt.Println("Main")
    done := make(chan bool)
    go do_stuff(done)
    <-done
}

Pourquoi <-done fonctionne-t-il ?

<-done est une opération de réception bloquante. Cela signifie que la routine principale attendra qu'une valeur soit envoyée au canal terminé. Puisque la goroutine do_stuff envoie explicitement une vraie valeur au canal lorsqu'elle est terminée, la routine principale la recevra et continuera l'exécution.

Que se passe-t-il si nous décommentons la dernière ligne ?

Si nous décommentons la dernière ligne, nous obtenons une erreur de blocage car le canal est vide et aucune autre goroutine ne lui envoie de valeurs. Par conséquent, la routine principale attendra indéfiniment qu'une valeur soit reçue, ce qui entraînera un blocage.

Approche alternative utilisant le package de synchronisation

Pour les scénarios de concurrence plus complexes, la synchronisation Le package fournit des mécanismes de synchronisation pratiques. Par exemple, si vous devez paralléliser une série de fonctions de longue durée, l'extrait de code suivant montre comment utiliser le type sync.WaitGroup :

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()
        }()
    }

    // will wait until wg.Done is called 10 times
    // since we made wg.Add(1) call 10 times
    wg.Wait()
}

func longOp() {
    time.Sleep(time.Second * 2)
    fmt.Println("long op done")
}

Dans cet exemple, WaitGroup garantit que la routine principale ne se termine pas tant que tous les goroutines n'ont pas appelé wg.Done.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!