Sobes.tech
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.