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这个方法的超时就不起作用了。这里暂且保留着吧。