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

出处:
https://studygolang.com/articles/5224

上周末参加Go技术聚会,京东的美女工程师讲到一个select-case和time.Ticker的使用注意事项(真实的应用场景是:在测试收包的顺序的时候,加了个tick就发现丢包了),觉得很有意思,记录一下。

package main

import (
    "fmt"
    "runtime"
    "time"
)

func init() {
    runtime.GOMAXPROCS(runtime.NumCPU())
}

func main() {
    ch := make(chan int, 1024)
    go func(ch chan int) {
        for {
            val := <-ch
            fmt.Printf("val:%d\n", val)
        }
    }(ch)

    tick := time.NewTicker(1 * time.Second)
    for i := 0; i < 20; i++ {
        select {
        case ch <- i:
        case <-tick.C:
            fmt.Printf("%d: case <-tick.C\n", i)
        }   

        time.Sleep(200 * time.Millisecond)
    }
    close(ch)
    tick.Stop()
}

输出如下:

val:0
val:1
val:2
val:3
val:4
val:5
6: case <-tick.C
val:7
val:8
val:9
10: case <-tick.C
val:11
val:12
val:13
val:14
15: case <-tick.C
val:16
val:17
val:18
val:19

问题出在这个select里面:

select {
    case ch <- i:
    case <-tick.C:
    fmt.Printf("%d: case <-tick.C\n", i)
}

当两个case条件都满足的时候,运行时系统会通过一个伪随机的算法决定哪个case将会被执行
所以当tick.C条件满足的那个循环,有某种概率造成ch<-i没有发送(虽然通道两端没有阻塞,满足发送条件)

解决方案1: 一旦tick.C随机的case被随机到,就多执行一次ch<-i (不体面,如果有多个case就不通用了)

select {
    case ch <- i:
    case <-tick.C:
    fmt.Printf("%d: case <-tick.C\n", i)
    ch <- i
}

解决方案2: 将tick.C的case单独放到一个select里面,并加入一个default(保证不阻塞)

select {
    case ch <- i:
}
select {
    case <-tick.C:
    fmt.Printf("%d: case <-tick.C\n", i)
    default:
}

两种解决方案的输出都是希望的结果:

val:0
val:1
val:2
val:3
val:4
5: case <-tick.C
val:5
val:6
val:7
val:8
val:9
10: case <-tick.C
val:10
val:11
val:12
val:13
val:14
15: case <-tick.C
val:15
val:16
val:17
val:18
val:19
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值