func main() {

ch := make(chan int, 1)
wg := sync.WaitGroup{}
count := 5
wg.Add(1) // Add 1 to the WaitGroup before starting the producer goroutine
go func(wg *sync.WaitGroup, ch chan int, count int) {
defer wg.Done()
for i := 0; i <= count; i++ {
ch <- i
}
}(&wg, ch, count)

go func() {
for {
select {
case v, ok := <-ch:
if !ok { // Check if the channel is closed
return
}
fmt.Println(v)
}
}
}()

wg.Wait()
close(ch)
fmt.Println("end")
}