Golang Stepik 3.9 Параллелизм ч.2 как решить?

Question

[des1roer]

https://stepik.org/lesson/345547/step/14?unit=329291

Вам необходимо написать функцию calculator следующего вида:

func calculator(arguments <-chan int, done <-chan struct{}) <-chan int
В качестве аргумента эта функция получает два канала только для чтения, возвращает канал только для чтения.

Через канал arguments функция получит ряд чисел, а через канал done - сигнал о необходимости завершить работу. Когда сигнал о завершении работы будет получен, функция должна в выходной (возвращенный) канал отправить сумму полученных чисел.

Функция calculator должна быть неблокирующей, сразу возвращая управление.

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

package main

import (
	"fmt"
)

func main() {
	arguments := make(chan int)
	done := make(chan struct{})
	result := calculator(arguments, done)
	for i := 0; i < 10; i++ {
		arguments <- 1
	}
	close(done)
	fmt.Println(<-result)
}

func calculator(arguments <-chan int, done <-chan struct{}) <-chan int {
	output := make(chan int)
	var x int

	go func(output chan int) {
		defer close(output)
		for {
			select { // Оператор select
			case <-arguments: // Ждет, когда проснется гофер
				x += <-arguments
				output <- x
				fmt.Println(x)
			case <-done: // Ждет окончания времени
				return
			}
		}

	}(output)

	return output
}

PS C:\proj> go build .\main.go; .\main.exe
1
2
3
4
5
0
PS C:\proj> go build .\main.go; .\main.exe
fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan send]:
main.main()
C:/proj/main.go:14 +0x6a

goroutine 6 [chan send]:
main.calculator.func1(0xc00005c120)
C:/proj/main.go:30 +0xcf
created by main.calculator
C:/proj/main.go:24 +0x10d

Answer 1

[falconandy]

package main

import (
	"fmt"
)

func main() {
	arguments := make(chan int)
	done := make(chan struct{})
	result := calculator(arguments, done)
	for i := 0; i < 10; i++ {
		arguments <- 1
	}
	close(done)
	fmt.Println(<-result)
}

func calculator(arguments <-chan int, done <-chan struct{}) <-chan int {
	output := make(chan int)

	go func(output chan int) {
		defer close(output)

		var x int
	Loop:
		for {
			select { // Оператор select
			case arg, ok := <-arguments: // Ждет, когда проснется гофер
				if !ok {
					break Loop
				}
				x += arg
				fmt.Println(x)
			case <-done: // Ждет окончания времени
				break Loop
			}
		}
		output <- x
	}(output)

	return output
}

Comments

[des1roer]

жесть какая. Loop в первый раз в жизни вижу ))))

[falconandy]

Т.к. мы не знаем, будут ли вообще читать из output канала, возможно стоит сделать его буферизированным
output := make(chan int, 1)
Канал arguments тоже может быть буферизированным - возможно стоит просуммировать накопленные в буфере значения при поступлении сигнала done

case <-done: // Ждет окончания времени
			Loop2:
				for {
					select {
					case arg, ok := <-arguments:
						if !ok {
							break Loop2
						}
						x += arg
						fmt.Println(x)
					default:
						break Loop2
					}
				}
				break Loop

[Александр Павлюк]

можно и без метки решить

[Everything_is_not_so_bad]

Александр Павлюк, тогда лучше дать свой вариант ответа.

[Александр Павлюк]

Román Mirilaczvili, например, так https://go.dev/play/p/xbwxvlPMccq

[falconandy]

Когда сигнал о завершении работы будет получен, функция должна в выходной (возвращенный) канал отправить сумму полученных чисел.

Чтобы соответствовать этому требованию, надо код решения немного переделать:

func calculator(arguments <-chan int, done <-chan struct{}) <-chan int {
	output := make(chan int)

	go func(output chan<- int) {
		defer close(output)

		var x int
		for {
			select { // Оператор select
			case arg, ok := <-arguments: // Ждет, когда проснется гофер
				if !ok { // Канал закрыт, больше из него читать не надо
					arguments = nil // Из nil канала чтение уже не сработает, select будет ждать только сигнала done
				} else {
					x += arg
					fmt.Println(x)
				}
			case <-done: // Ждет окончания времени
				output <- x
				return
			}
		}
	}(output)

	return output
}

[Максим Федоров]

des1roer, это goto, везде есть :)