Sobes.tech
Senior

Co říct interviewérovi (pokud se zeptají) "Použil bych worker pool s pevně stanoveným počtem workerů, kanál pro úkoly a sync.WaitGroup. Workeři budou provádět požadavky, a timeout a zrušení budou řízeny přes kontext. Výsledky shromáždím v bufferu, chráněném mutexem (nebo přes samostatný kanál)." Teď je řada na tobě Napiš kód v tomto chatu. Já ho zkontroluji a dám zpětnou vazbu. Pokud nevíš, kde začít, požádej o nápovědu. Hodně štěstí!

sobes.tech AI

Odpověď od AI

go package main

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

// Task představuje úlohu ke zpracování type Task struct { id int // další pole }

// WorkerPool implementuje skupinu pracovníků func WorkerPool(ctx context.Context, tasks <-chan Task, results chan<- string, wg *sync.WaitGroup) { defer wg.Done() for { select { case <-ctx.Done(): return case task, ok := <-tasks: if !ok { return } // Zpracování úlohy res := processTask(task) select { case results <- res: case <-ctx.Done(): return } } } }

func processTask(t Task) string { // simulace práce time.Sleep(100 * time.Millisecond) return fmt.Sprintf("výsledek pro úlohu %d", t.id) }

func main() { tasks := make(chan Task) results := make(chan string) var wg sync.WaitGroup

ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()

workerCount := 5
wg.Add(workerCount)
for i := 0; i < workerCount; i++ {
	go WorkerPool(ctx, tasks, results, &wg)
}

// Samostatná goroutine pro uzavření kanálu výsledků po dokončení pracovníků
go func() {
	wg.Wait()
	close(results)
}()

// Odesílání úloh
go func() {
	for i := 0; i < 20; i++ {
		tasks <- Task{id: i}
	}
	close(tasks)
}()

// Sbírání výsledků
for res := range results {
	fmt.Println(res)

} }