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

1、没有超时的时候

package main

import (
"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 main

import (
"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)
}