Я пытаюсь реализовать простой пример распараллеливания в 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!
Проблема в том, что range
по каналу продолжает пытаться получить от канала, пока канал не будет закрыт, но вы не close
каналы после завершения отправки.
Добавьте close(out)
перед wg.Done
в вашем коде, и это исправит.
Детская площадка: https://play.golang.org/p/NbKTx6Lke7X
Обновлено: исправлена ошибка закрытия закрытого канала.