[golang]time.After 在select中使用的正确姿势（解决超时逻辑未

前言

select 的语法如下所示

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

问题复现

package main

import(
    "fmt"
    "time"
)

func add(ch chan int) {
    for i:=0;i<10;i++{
        ch <- i
    }
}

// timeout problem recurrent
func 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...")
        }
    }
}

根据条件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 随机性失败
当case <- ch 和 case <- ticker.C 同时成立时，Select会随机公平地选出一个执行，有可能选择到前者，导致超时随机行失败

最终解决方式

// final solution
func test4() {
    ticker := time.NewTicker(2 * time.Second)
    defer ticker.Stop()
    ch := make(chan int,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
        }
    }
}

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

参考文档

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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!