Предотвращение паники при делении на ноль

Условие

Необходимо изменить программу так, чтобы при попытке деления на ноль не возникала паника. Важно корректно синхронизировать горутины и гарантировать, что ресурс канала освобождается независимо от результата деления.

package main

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

func main() {
    wg := sync.WaitGroup{}
    ch := make(chan struct{}, 5)

    for i := 0; i < 10; i++ {
        i := i
        wg.Add(1)
        go func() {
            defer wg.Done()
            printNumber(ch, i)
        }()
    }

    wg.Wait()
    close(ch)
}

func printNumber(ch chan struct{}, n int) {
    ch <- struct{}{}
    time.Sleep(time.Second)
    if n != 0 {
        fmt.Println(1 / n)
    }
    <-ch
}

Ответ

Деление на ноль уже исключено условием n != 0. Освобождение слота семафора лучше перенести в defer сразу после успешного захвата.

Меняется только printNumber; WaitGroup в main уже ожидает завершения всех работников перед закрытием канала.

func printNumber(ch chan struct{}, n int) {
    ch <- struct{}{}
    defer func() { <-ch }()
    time.Sleep(time.Second)
    if n == 0 {
        return
    }
    fmt.Println(1 / n)
}

При целочисленном делении для n = 1 выводится 1, для 2–9 — 0; порядок конкурентных выводов не фиксирован. defer освобождает слот и при раскрутке стека из-за другой паники, но не подавляет её. recover для ожидаемого нулевого аргумента не нужен: проверка явнее и безопаснее.