Concurrence en Go

Go Worker Pool

Share a queue of jobs between a fixed number of goroutines to limit how much runs at once.

  • Avancé
  • 4 min de lecture
  • 1 exemple exécutable
Ce que vous allez apprendre
  • A jobs channel
  • Several workers
  • Closing results safely
6 jobs, 3 workers
  1. 1jobs1 2 3 4 5 6
  2. 23 workersn * n
  3. 3results1 4 9 16 25 36

A fixed team of workers

A worker pool starts a fixed number of goroutines that all read jobs from one channel. It limits how much work runs at the same time, which protects memory, files and network connections.

Three workers read from jobs until it is closed and send each answer to results. A separate goroutine closes results after every worker has finished, which ends the final range loop. Each result carries its job number, so the order in which workers finish does not matter.

Points clés

  1. All workers read from one jobs channel.
  2. close(jobs) tells them to stop.
  3. Results are closed after every worker is done.

Exemple Go exécutable

Go
Exécuter le code →
main.go
package main

import (
	"fmt"
	"sync"
)

func worker(jobs <-chan int, results chan<- [2]int, wg *sync.WaitGroup) {
	defer wg.Done()
	for n := range jobs {
		results <- [2]int{n, n * n}
	}
}

func main() {
	jobs := make(chan int)
	results := make(chan [2]int)
	var wg sync.WaitGroup

	for w := 0; w < 3; w++ {
		wg.Add(1)
		go worker(jobs, results, &wg)
	}

	go func() {
		for n := 1; n <= 6; n++ {
			jobs <- n
		}
		close(jobs)
	}()
	go func() {
		wg.Wait()
		close(results)
	}()

	squares := make([]int, 7)
	for result := range results {
		squares[result[0]] = result[1]
	}
	fmt.Println(squares[1:])
}
Obtenir un bouton Run pour cet exemple Résultat de l'exécution
[1 4 9 16 25 36]

S'entraîner

Ouvrez l'exemple dans le compilateur en ligne, modifiez une valeur ou ajoutez une ligne, et devinez la sortie avant de l'exécuter. Cassez ensuite le code exprès et lisez le message du compilateur.

Leçon terminée ?

Marquez-la comme apprise pour suivre votre progression.

S'entraîner avec un défi

Utilisez le contenu de cette leçon pour résoudre un vrai problème. Chaque défi s'exécute dans le navigateur, avec des tests instantanés et une revue par l'IA.

Tous les défis