用go的goroutine和channel实现一个简单的“生产、消费”(带有超时控制)小例子 - Go语言中文社区

用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)
}

执行结果
这里写图片描述
5秒之后超时(此时队列里尚有未处理完的数据)。

版权声明:本文来源CSDN,感谢博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
原文链接:https://blog.csdn.net/u010230794/article/details/78626188
站方申明:本站部分内容来自社区用户分享,若涉及侵权,请联系站方删除。
  • 发表于 2019-09-14 00:09:47
  • 阅读 ( 1451 )
  • 分类:Go

0 条评论

请先 登录 后评论

官方社群

GO教程

猜你喜欢