Senior
/** * Merge N channels into one * * Given n channels of type chan int. Write a function that merges all data **/ // 1. Implement a function to merge N channels into one func joinChannels(chs ...<-chan int) <-chan int { mergedCh := make(chan int) // ... return mergedCh } // 2. Write code to fill channels a, b, c with values // ...
sobes.tech AI
Answer from AI
go func joinChannels(chs ...<-chan int) <-chan int { mergedCh := make(chan int)
var wg sync.WaitGroup
wg.Add(len(chs))
for _, ch := range chs {
go func(c <-chan int) {
defer wg.Done()
for val := range c {
mergedCh <- val
}
}(ch)
}
go func() {
wg.Wait()
close(mergedCh)
}()
return mergedCh
}
// Example of filling channels a, b, c func main() { a := make(chan int, 3) b := make(chan int, 3) c := make(chan int, 3)
a <- 1
a <- 2
a <- 3
close(a)
b <- 4
b <- 5
b <- 6
close(b)
c <- 7
c <- 8
c <- 9
close(c)
merged := joinChannels(a, b, c)
for val := range merged {
fmt.Println(val)
}
}
In this example, the `joinChannels` function launches a goroutine for each input channel, reading from it and sending data to the shared `mergedCh`. Once all source channels are closed and processed, the shared channel is closed. This allows merging multiple data streams into a single channel for easy reading.