Αποστολή Merge channel Υλοποιήστε τη λειτουργία merge, η οποία δέχεται έναν αυθαίρετο αριθμό καναλιών chan int και επιστρέφει ένα έξοδο κανάλι, από το οποίο μπορείτε να διαβάσετε όλες τις τιμές των μεταβιβασμένων καναλιών. Τα μεταβιβασμένα κανάλια εκτός της λειτουργίας merge κλείνουν μετά την εγγραφή. Μετά την ολοκλήρωση της ανάγνωσης από το συγχωνευμένο κανάλι, το πρόγραμμα πρέπει να εμφανίσει "ok".

Ниже пример реализации функции merge, которая принимает произвольное количество каналов chan int и возвращает один канал, из которого можно читать все значения. После закрытия всех входных каналов выходной канал тоже закрывается, и программа выводит "ok".

package main

import (
	"fmt"
	"sync"
)

func merge(channels ...chan int) chan int {
	out := make(chan int)
	var wg sync.WaitGroup

	wg.Add(len(channels))

	for _, ch := range channels {
		go func(c chan int) {
			defer wg.Done()
			for v := range c {
				out <- v
			}
		}(ch)
	}

	go func() {
		wg.Wait()
		close(out)
	}()

	return out
}

func main() {
	ch1 := make(chan int)
	ch2 := make(chan int)

	go func() {
		defer close(ch1)
		ch1 <- 1
		ch1 <- 2
	}()

	go func() {
		defer close(ch2)
		ch2 <- 3
		ch2 <- 4
	}()

	merged := merge(ch1, ch2)

	for v := range merged {
		fmt.Println(v)
	}

	fmt.Println("ok")
}

В этом примере:

• Для каждого входного канала запускается горутина, которая читает из него и отправляет значения в общий канал out.
• Используется sync.WaitGroup для ожидания завершения чтения из всех каналов.
• После того, как все входные каналы закрыты и данные прочитаны, выходной канал закрывается.
• В main после чтения всех значений выводится "ok".