Простой параллельный пример с каналом, ведущий к взаимоблокировке

Я пытаюсь реализовать простой пример распараллеливания в golang с использованием каналов. Код пытается реализовать параллельную функцию карты. Каналы также буферизированы для ослабления ограничений на блокировку канала. но код все равно ведет в тупик.

func pmmap(inp []int, f func(int) int, p int) {

    var wg sync.WaitGroup
    var output []chan int
    tot := len(inp)
    out := make(chan int, tot)
    slice := tot / p

    for i := 0; i < p; i++ {
        temp := make(chan int, slice)
        output = append(output, temp)
        start_ind := slice * i
        end_ind := start_ind + slice
        fmt.Println(start_ind, end_ind)

        wg.Add(1)
        go func(si, ei int, out chan int, wg *sync.WaitGroup) {
            fmt.Println("goroutine started with ", si, ei)
            for ind := si; ind < ei; ind++ {
                out <- f(inp[ind])
            }
            wg.Done()
        }(start_ind, end_ind, output[i], &wg)
    }

    wg.Add(1)
    go func(wg *sync.WaitGroup) {
        for i := 0; i < p; i++ {
            for val := range output[i] {
                out <- val
            }
            close(output[i])
        }
        wg.Done()
    }(&wg)

    wg.Add(1)
    go func(wg *sync.WaitGroup) {
        for i := range out {
            fmt.Println(i)
        }
        wg.Done()
    }(&wg)

    time.Sleep(time.Second * 6)
    wg.Wait()
    close(out)
}

func add4(i int) int {
    return i + 4
}

func main() {
    temp := []int{}
    for i := 1; i <= 20; i++ {
        temp = append(temp, i)
    }
    pmmap(temp, add4, 2)
}

из вывода приведенного выше кода я получаю тупик, потому что вывод канала [1] никогда не читается. но я не уверен, почему

0 10
10 20
goroutine started with  0 10
goroutine started with  10 20
5
6
7
8
9
10
11
12
13
14
fatal error: all goroutines are asleep - deadlock!
Создание API ввода вопросов на разных языках программирования (Python, PHP, Go и Node.js)
Создание API ввода вопросов на разных языках программирования (Python, PHP, Go и Node.js)
API ввода вопросов - это полезный инструмент для интеграции моделей машинного обучения, таких как ChatGPT, в приложения, требующие обработки...
0
0
60
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Проблема в том, что range по каналу продолжает пытаться получить от канала, пока канал не будет закрыт, но вы не close каналы после завершения отправки.

Добавьте close(out) перед wg.Done в вашем коде, и это исправит.

Детская площадка: https://play.golang.org/p/NbKTx6Lke7X

Обновлено: исправлена ​​ошибка закрытия закрытого канала.

Другие вопросы по теме