select 多路选择

select写法上跟switch case的写法基本一致,只不过golang的select是通信控制语句。select的执行必须有通信的发送或者接受,如果没有就一直阻塞。

	ch := make(chan bool, 0)
	ch1 := make(chan bool, 0)
	select {
		case ret := <-ch:
			fmt.Println(ret)
		case ret := <-ch1:
			fmt.Println(ret)
	}

如果ch和ch1都没有通信数据发送,select就一直阻塞,直到ch或者ch1有数据发送,select就执行相应的case来接受数据。

select 实现超时控制

我们可以利用select机制实现一种简单的超时控制。
先看下程序完整执行的代码

func service(ch chan bool) {
	time.Sleep(time.Second*3)
	ch<-true
}
func main() {
	ch := make(chan bool, 0)
	go service(ch)
	select {
		case ret := <-ch:
			fmt.Println(ret)
		case <-time.After(time.Second*5):
			fmt.Println("timeout")
	}
}

___go_build_main_go #gosetup
true

可以看到使用time.After超时定义了5S,service程序执行3S,所以肯定没有超时,跟预想的一致。
我们再看看超时的执行,我们将service程序执行时间该为6S。超时控制继续是5S,再看下执行效果

func service(ch chan bool) {
	time.Sleep(time.Second*6)
	ch<-true
}
func main() {
	ch := make(chan bool, 0)
	go service(ch)
	select {
		case ret := <-ch:
			fmt.Println(ret)
		case <-time.After(time.Second*5):
			fmt.Println("timeout")
	}
}

___go_build_main_go #gosetup
timeout

执行到了超时的case,跟预想的其实是一致的。

select 判断channel是否关闭

先看下接受数据的语法

val,ok <- ch
ok true 正常接收数据
ok false 通道关闭

可以看到接受数据其实有两个参数,第二个bool值会反应channel是否关闭,是否可以正常接受数据。

看下测试代码
我们写了一个数据发送者,两个数据接收者,当发送者关闭channel的时候,两个接收者的 goroutine 可以通过以上的语法判断channel是否关闭,决定自己的 goroutine 是否结束。

func sender(ch chan int, wg *sync.WaitGroup) {
	for i:=0;i<10;i++ {
		ch<-i
	}
	close(ch)
	wg.Done()
}
func receiver(ch chan int, wg *sync.WaitGroup) {
	for {
		if val,ok := <-ch;ok {
			fmt.Println(fmt.Sprintf("%d,%s",val, "revevier"))
		} else {
			fmt.Println("quit recevier")
			break;
		}
	}
	wg.Done()
}
func receiver2(ch chan int, wg *sync.WaitGroup) {
	for {
		if val,ok := <-ch;ok {
			fmt.Println(fmt.Sprintf("%d,%s",val, "revevier2"))
		} else {
			fmt.Println("quit recevier2")
			break;
		}
	}
	wg.Done()
}
func main() {
	ch := make(chan int, 0)
	wg := &sync.WaitGroup{}
	wg.Add(1)
	go sender(ch, wg)
	wg.Add(1)
	go receiver(ch, wg)
	wg.Add(1)
	go receiver2(ch, wg)
	wg.Wait()
}

执行结果

0,revevier2
2,revevier2
3,revevier2
4,revevier2
5,revevier2
6,revevier2
7,revevier2
1,revevier
9,revevier
quit recevier
8,revevier2
quit recevier2

可以看到一个数据发送者,两个数据接收者,当channel关闭的时候,两个数据接收者都收到了channel关闭的通知。
需要注意的是,给一个已经关闭的channel发送数据,程序会panic,从一个已经关闭的channel接收数据,会接收到没有参考意义的channel类型的0值数据,Int是0,string是空...

select 退出计时器等程序

开发中经常会经常会使用轮训计时器,但是当程序退出时,轮训计时器无法关闭的问题。其实select是可以解决这个问题的。
如果我们有一个轮训任务,需要一个timer,每隔3S执行逻辑,过完10S之后关闭这个timer。

看下代码

func TimeTick(wg *sync.WaitGroup,q chan bool) {
	defer wg.Done()
	t := time.NewTicker(time.Second*3)
	defer t.Stop()
	for {
		select {
		case <-q:
			fmt.Println("quit")
			return
		case <-t.C:
			fmt.Println("seconds timer")
		}
	}
}
func main() {
	q := make(chan bool)
	wg := new(sync.WaitGroup)
	wg.Add(1)
	go TimeTick(wg,q)
	time.Sleep(time.Second*10)
	close(q)
	wg.Wait()
}

执行结果

seconds timer
seconds timer
seconds timer
quit

很优雅的通过关闭channel退出了轮训计时器 goroutine,

09-26 12:24