Snippet

Implementierung eines Worker-Pools zur parallelen Aufgabenverarbeitung

Ein Worker-Pool verwaltet eine Gruppe von concurrent laufenden Goroutines, die Aufgaben aus einer gemeinsamen Warteschlange abarbeiten. Dieses Muster verhindert unkontrollierte Goroutine-Erstellung und Ressourcenerschöpfung. Wir verwenden ein sync.WaitGroup, um das Ende der Worker zu verfolgen, schließen den Job-Kanal, um den Workern das Beenden zu signalisieren, und schließen schließlich den Ergebnis-Kanal sicher, sobald alle Worker fertig sind.

snippet.go

package main
 
import (
	"fmt"
	"sync"
)
 
func worker(id int, jobs <-chan int, results chan<- int, wg *sync.WaitGroup) {
	defer wg.Done()
	for job := range jobs {
		results <- job * 2
	}
}
 
func runWorkerPool(numWorkers int, numJobs int) []int {
	jobs := make(chan int, numJobs)
	results := make(chan int, numJobs)
	var wg sync.WaitGroup
 
	for w := 1; w <= numWorkers; w++ {
		wg.Add(1)
		go worker(w, jobs, results, &wg)
	}
 
	for j := 1; j <= numJobs; j++ {
		jobs <- j
	}
	close(jobs)
 
	wg.Wait()
	close(results)
 
	var output []int
	for res := range results {
		output = append(output, res)
	}
	return output
}

Erklärung

func worker(id int, jobs <-chan int, results chan<- int, wg *sync.WaitGroup)

Definiert einen Worker, der aus dem Nur-Lese-Job-Kanal liest und in den Nur-Schreib-Ergebnis-Kanal schreibt.

wg.Add(1)

Erhöht den WaitGroup-Zähler für jede gestartete Worker-Goroutine.

close(jobs)

Schließt den Job-Kanal, was signalisiert, dass keine Arbeit mehr gesendet wird; dies beendet die range jobs-Schleife.

wg.Wait()

Blockiert, bis alle Worker-Goroutines wg.Done() aufrufen, um sicherzustellen, dass alle Aufgaben verarbeitet sind.

Vorheriges Snippet Nächstes Snippet

Mehr Go-Übungen für Fortgeschrittene →

Implementierung eines Worker-Pools zur parallelen Aufgabenverarbeitung

Verwandt