52 lines
544 B
Go
52 lines
544 B
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
)
|
|
|
|
func producer(ch chan int) {
|
|
defer close(ch)
|
|
for i := 0; i < 5; i++ {
|
|
ch <- i
|
|
}
|
|
}
|
|
|
|
func consumer(ch chan int) {
|
|
|
|
/*
|
|
for {
|
|
select {
|
|
case value, opened := <-ch:
|
|
if !opened {
|
|
return
|
|
}
|
|
|
|
fmt.Println(value)
|
|
}
|
|
}
|
|
*/
|
|
|
|
for value := range ch { // syntax sugar
|
|
fmt.Println(value)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
ch := make(chan int)
|
|
wg := sync.WaitGroup{}
|
|
wg.Add(2)
|
|
|
|
go func() {
|
|
defer wg.Done()
|
|
producer(ch)
|
|
}()
|
|
|
|
go func() {
|
|
defer wg.Done()
|
|
consumer(ch)
|
|
}()
|
|
|
|
wg.Wait()
|
|
}
|