用go的goroutine和channel实现一个简单的“生产、消费”（带有超时控制）小例子

直接上代码

1、没有超时的时候

package mainimport (    "fmt"    "time")func main() {    //存放生产的channel    jobChan := make(chan int, 100)    //通知是否完成所有任务    endChan := make(chan bool)    go production(jobChan)    go worker(jobChan, endChan)    select {    case <-endChan:        fmt.Println("消费完成……………………")        return    case <-time.After(time.Second * 20):        fmt.Println("超时………………………")        return    }}//消费func worker(jobChan <-chan int, endChan chan bool) {    for job := range jobChan {        fmt.Println("消费:", job)    }    //消费结束，通知endChan    endChan <- true}//生产func production(jobChan chan<- int) {    for i := 1; i <= 10; i++ {        fmt.Println("生产:", i)        jobChan <- i    }    //关闭channel防止消费阻塞    close(jobChan)}

执行结果 ：

可见生产和消费异步并发执行。

2、当超时时 （timeout设置小点，并增加消费处理时间）

package mainimport (    "fmt"    "time")func main() {    //存放生产的channel    jobChan := make(chan int, 100)    //通知是否完成所有任务    endChan := make(chan bool)    go production(jobChan)    go worker(jobChan, endChan)    select {    case <-endChan:        fmt.Println("消费完成……………………")        return    case <-time.After(time.Second * 5):        fmt.Println("超时………………………")        return    }}//消费func worker(jobChan <-chan int, endChan chan bool) {    for job := range jobChan {        fmt.Println("消费:", job)        //增加消费处理时间        time.Sleep(1 * time.Second)    }    //消费结束，通知endChan    endChan <- true}//生产func production(jobChan chan<- int) {    for i := 1; i <= 10; i++ {        fmt.Println("生产:", i)        jobChan <- i    }    //关闭channel防止消费阻塞    close(jobChan)}

执行结果

5秒之后超时（此时队列里尚有未处理完的数据）。