Middle+
Задача: „Паралелна обработка на URL с ограничение по време“ Напишете функция fetchAll(urls []string, maxParallel int, timeout time.Duration) map[string]string, която: 1. За всяка URL изпълнява HTTP GET заявка (може да се използва http.Get). 2. Връща map[string]string, където ключът е URL, а стойността е тялото на отговора (като низ) или празен низ при грешка/тайм-аут. 3. Не се изпълняват повече от maxParallel заявки едновременно. 4. Общото време за работа на функцията не трябва да надвишава timeout. Ако изтече времето, всички незавършени заявки се прекъсват, и функцията връща това, което е успяла. 5. Използвайте контекст за анулиране.
sobes.tech AI
Отговор от AI
go package main
import ( "context" "io/ioutil" "net/http" "sync" "time" )
func fetchAll(urls []string, maxParallel int, timeout time.Duration) map[string]string { ctx, cancel := context.WithTimeout(context.Background(), timeout) defer cancel()
results := make(map[string]string)
var mu sync.Mutex
sem := make(chan struct{}, maxParallel) // семафор за ограничување на паралелизмот
var wg sync.WaitGroup
for _, url := range urls {
select {
case <-ctx.Done():
break
default:
wg.Add(1)
sem <- struct{}{} // зазема место
go func(u string) {
defer wg.Done()
defer func() { <-sem }() // ослободува место
req, err := http.NewRequestWithContext(ctx, "GET", u, nil)
if err != nil {
mu.Lock()
results[u] = ""
mu.Unlock()
return
}
resp, err := http.DefaultClient.Do(req)
if err != nil {
mu.Lock()
results[u] = ""
mu.Unlock()
return
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
mu.Lock()
results[u] = ""
mu.Unlock()
return
}
mu.Lock()
results[u] = string(body)
mu.Unlock()
}(url)
}
}
wg.Wait()
return results
}