基于go的生产者消费者模型

基于go的生产者消费者模型

//生产者
func Producer(ch chan int) {
    for i := 1; i <= 10; i++ {
        ch <- i
    }
    close(ch)
}

//消费者
func Consumer(id int, ch chan int, done chan bool) {
    for {
        value, ok := <-ch
        if ok {
            fmt.Printf("id : %d consum %d value \n", id, value)
        } else {
            fmt.Printf("id : %d had closed \n", id)
            break
        }
    }
    done <- true
}

测试文件

const (
    cur int = 2
)

func main() {
    ch := make(chan int, 3)
    done := make(chan bool, 1)
    for i := 1; i <= 2; i++ {
        go Consumer(i, ch, done)
    }
    go Producer(ch)
        //注意这里，如果done channel里面没有true的化会一直阻塞直到有东西进来。
    for i := 1; i <= cur; i++ {
        <-done
    }
}