go语言基于time.After通道超时设计和通道关闭close

go语言中多个并发程序的数据同步是采用通道来传输,比如v:=<-chan,从通道里读取数据到v,是一个阻塞操作。可是如通道里没有数据写入,就是chan<-data,这样写入通道的操作,在读操作时就会一直阻塞,需要加入一个超时机制来进行判断。

具体的超时设计是通过使用select和case语句,类似于switch和case,在每一个case里进行一个io操作,比如读或者写,在最后一个case里调用time包里的After方法,可以达到超时检测效果。参考下面例子1

当然,如写入端在写入通道结束后,调用close(chan)关闭通道。在读取端,就会读到一个该通道类型的空值,如是int就是0,如是string就是""空字符串,可以根据这个空值来判断,或者使用两个返回值来读取通道:v,br:=<-chan,这里第2个参数br是一个bool变量,表示通道是否关闭。参考下面例子2

例子1如下:

package main

import (
	"fmt"
	"time"
)

func main() {
	ch := make(chan string, 2)//定义了缓冲长度2的通道,类型是字符串,可以连续写入2次数据
	go func(c chan string) {
		for i := 0; i < 3; i++ {
			str := fmt.Sprintf("%d", i)
			c <- str
			time.Sleep(time.Millisecond * 10)
		}
	}(ch)
	go func(c chan string) {
		for i := 10; i < 13; i++ {
			str := fmt.Sprintf("%d", i)
			c <- str
			time.Sleep(time.Millisecond * 10)
		}
	}(ch)
	timelate := 0 //定义超时次数
	for {
		time.Sleep(time.Millisecond * 2000) //每隔2秒读取下管道
		select {
		case i := <-ch:
			fmt.Println("通道读取到:", i)
		case <-time.After(time.Second * 2): // 等待2秒超时,这里time.After 返回一个只读通道,就是当前时间值
			timelate++
			fmt.Printf("通道接收超时,第%d次\n", timelate)
			if timelate > 2 {
				goto end
			}
		}
	}
end:
	fmt.Println("退出88")
}

例子2,演示了close关闭通道,使用2个返回值来读取通道,获取通道关闭状态。

package main

import (
	"fmt"
	"time"
)

func main() {
	ch := make(chan string, 2) //定义了缓冲长度2的通道,类型是字符串,可以连续写入2次数据
	go func(c chan string) {
		for i := 0; i < 3; i++ {
			str := fmt.Sprintf("%d", i)
			c <- str
			time.Sleep(time.Millisecond * 10)
		}
	}(ch)
	go func(c chan string) {
		for i := 10; i < 13; i++ {
			str := fmt.Sprintf("%d", i)
			c <- str
			time.Sleep(time.Millisecond * 10)
		}
		time.Sleep(time.Millisecond * 1000) //专门给这个协程加个1秒的延时,让它晚退出会,好调用close关闭通道。
		close(c)
	}(ch)
	timelate := 0 //定义超时次数
	for {
		time.Sleep(time.Millisecond * 2000) //每隔2秒读取下管道
		select {
		case i, br := <-ch: //从通道里读取2个返回值,第2个是通道是否关闭的bool变量
			if !br { //如果是false,表示通道关闭
				fmt.Println("通道关闭了")
				goto end
			}
			fmt.Println("通道读取到:", i)
		case <-time.After(time.Second * 2): // 等待2秒超时,这里time.After 返回一个只读通道,就是当前时间值
			timelate++
			fmt.Printf("通道接收超时,第%d次\n", timelate)
			if timelate > 2 {
				goto end
			}
		}
	}
end:
	fmt.Println("退出88")
}

对于例子2来说,这里因为在通道写入端用close关闭通道了,所以case <-time.After这个方法的超时就不起作用了。这里暂且保留着吧。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值