Maison >développement back-end >Golang >Programmation simultanée Golang : utiliser Go WaitGroup pour implémenter le planificateur de tâches
Programmation simultanée Golang : utilisez Go WaitGroup pour implémenter le planificateur de tâches
WaitGroup maintient un compteur en interne pour enregistrer le nombre de tâches inachevées. Lorsque la méthode Add est appelée, le compteur est incrémenté de la valeur spécifiée ; lorsque la méthode Done est appelée, le compteur est décrémenté de 1 lorsque la méthode Wait est appelée, si le compteur est supérieur à 0, la goroutine actuelle sera bloqué jusqu'à ce que le compteur revienne à zéro.
package main import ( "fmt" "sync" "time" ) func worker(id int, wg *sync.WaitGroup) { defer wg.Done() fmt.Printf("Worker %d starting ", id) time.Sleep(time.Second) fmt.Printf("Worker %d done ", id) } func main() { var wg sync.WaitGroup for i := 1; i <= 5; i++ { wg.Add(1) go worker(i, &wg) } wg.Wait() fmt.Println("All workers done") }
Dans cet exemple, nous utilisons une boucle for pour démarrer 5 goroutines de travail et appelons wg.Add(1) pour ajouter le nombre de tâches au WaitGroup. Dans la fonction de travail, utilisez le mot-clé defer pour vous assurer que wg.Done() est appelé avant que la fonction ne revienne pour décrémenter la valeur du compteur. Enfin, utilisez wg.Wait() pour attendre que toutes les tâches soient terminées.
Worker 1 starting Worker 2 starting Worker 3 starting Worker 4 starting Worker 5 starting Worker 3 done Worker 1 done Worker 2 done Worker 5 done Worker 4 done All workers done
Vous pouvez voir que toutes les goroutines de travail sont exécutées simultanément, mais l'ordre de sortie n'est pas nécessairement dans l'ordre de démarrage. la planification est effectuée par le runtime Go.
Grâce à l'introduction et à l'exemple de code de cet article, je pense que vous comprenez mieux comment utiliser WaitGroup de Golang pour implémenter un planificateur de tâches. J'espère que cet article vous aidera à apprendre et à utiliser la programmation simultanée dans Golang !
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!