Principal
// Lors de la mise à jour d'une commande, nous devons envoyer les données de la commande à plusieurs services (tiers) // le nombre de services augmente (peut atteindre des milliers+) // nous avons écrit le code, au début tout allait bien, mais avec le temps notre service a commencé à consommer beaucoup de ressources // Il faut optimiser ce code pour qu'il fonctionne efficacement avec un grand nombre de services func (s *orderService) SendOrder(ctx context.Context, hosts []string, order Order) { for i := 0; i < len(hosts); i++ { go func() { // Imaginez que c'est un appel réseau long response, err := s.httpClient.Send(ctx, hosts[i], order) if err != nil { s.logger.Error(ctx, "échec de l'envoi", err) return } s.logger.Info(ctx, "succès", response) }() } }
sobes.tech IA
Réponse de l'IA
В представленном коде есть несколько проблем, которые приводят к высокому потреблению ресурсов при большом количестве сервисов:
-
Потенциальная утечка переменной цикла: в анонимной функции используется
hosts[i], ноiможет измениться к моменту выполнения горутины. Нужно передаватьhosts[i]как параметр функции. -
Запуск слишком большого количества горутин одновременно: если
hostsсодержит тысячи элементов, будет создано столько же горутин, что приведёт к высокому потреблению памяти и CPU. -
Отсутствие контроля параллелизма и ожидания завершения.
Как оптимизировать:
- Использовать пул горутин с ограничением по количеству одновременно выполняемых задач (например, через семафор или канал).
- Передавать значение
hostв горутину явно, чтобы избежать захвата переменной цикла. - Добавить механизм ожидания завершения всех отправок (например,
sync.WaitGroup).
Пример оптимизированного кода:
func (s *orderService) SendOrder(ctx context.Context, hosts []string, order Order) {
const maxConcurrency = 100 // ограничение по числу параллельных запросов
sem := make(chan struct{}, maxConcurrency)
var wg sync.WaitGroup
for _, host := range hosts {
wg.Add(1)
sem <- struct{}{} // захватываем слот
go func(h string) {
defer wg.Done()
defer func() { <-sem }() // освобождаем слот
response, err := s.httpClient.Send(ctx, h, order)
if err != nil {
s.logger.Error(ctx, "failed to send", err)
return
}
s.logger.Info(ctx, "success", response)
}(host)
}
wg.Wait() // ждем завершения всех горутин
}
Такой подход позволяет контролировать нагрузку на систему, не создавая слишком много горутин одновременно, и предотвращает утечку переменной цикла.