[golang]time.After 在select中使用的正确姿势(解决超时逻辑未生效bug)

前端之家收集整理的这篇文章主要介绍了[golang]time.After 在select中使用的正确姿势(解决超时逻辑未生效bug)前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

前言

select 的语法如下所示

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

import(
    "fmt"
    "time"
)

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

@H_301_47@// 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) @H_301_47@// 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) @H_301_47@// 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会随机公平地选出一个执行,有可能选择到前者,导致超时随机行失败

最终解决方

@H_301_47@// 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: @H_301_47@// forbid block
        }
        select {
            case <- ticker.C:
                fmt.Println("timeout")
                return
            default: @H_301_47@// forbid block
        }
    }
}

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

参考文档

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

原文链接:https://www.f2er.com/go/188976.html

猜你在找的Go相关文章