time.After 在select-case中使用的正确姿势

前言

select 的语法如下所示

每个case都必须是一个通信所有channel表达式都会被求值所有被发送的表达式都会被求值如果任意某个通信可以进行，它就执行；其他被忽略。如果有多个case都可以运行，Select会随机公平地选出一个执行。其他不会执行。 否则：如果有default子句，则执行该语句。如果没有default字句，select将阻塞，直到某个通信可以运行；Go不会重新对channel或值进行求值。
1
2
3
4
5
6
7

问题复现

package mainimport(    "fmt"    "time")func add(ch chan int) {    for i:=0;i<10;i++{        ch <- i    }}// timeout problem recurrentfunc test2() {    ch := make(chan int, 10)    go add(ch)    for {        select {            case <- time.After(2 * time.Second):                fmt.Println("timeout")                return            case <- ch:                fmt.Println(ch) // if ch not empty, time.After will nerver exec                fmt.Println("sleep one seconds ...")                time.Sleep(1 * time.Second)                fmt.Println("sleep one seconds end...")        }    }}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

根据条件5：如果有多个case都可以运行，Select会随机公平地选出一个执行。其他不会执行。 
但是运行上述代码，当ch通道中存在数据时，time.After总是得不到运行，因此到时超时未生效（就像是两个case都成立时，select 都”公平”地选择了 case <- ch，导致超时逻辑未生效）

改进1

func test3() {    ticker := time.NewTicker(2 * time.Second)    defer ticker.Stop()    ch := make(chan int, 10)    go add(ch)    for {        select {            case <- ch:                fmt.Println(ch) // if ch not empty, time.After will nerver exec                fmt.Println("sleep one seconds ...")                time.Sleep(1 * time.Second)                fmt.Println("sleep one seconds end...")            case <- ticker.C:                fmt.Println("timeout")                return            default:        }    }}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

改进1 随机性失败 
当case <- ch 和 case <- ticker.C 同时成立时，Select会随机公平地选出一个执行，有可能选择到前者，导致超时随机行失败

最终解决方式

// final solutionfunc test4() {    ticker := time.NewTicker(2 * time.Second)    defer ticker.Stop()    ch := make(chan int, 10)    go add(ch)    for {        select {            case <- ch:                fmt.Println(ch) // if ch not empty, time.After will nerver exec                fmt.Println("sleep one seconds ...")                time.Sleep(1 * time.Second)                fmt.Println("sleep one seconds end...")            default: // forbid block        }        select {            case <- ticker.C:                fmt.Println("timeout")                return            default: // forbid block        }    }}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

将【超时】和【收包】放在各自单独的select里面，【超时】一定可以执行到

参考文档

go里面select-case和time.Ticker的使用注意事项 
go语言time包用法