方式2

第二个技巧的源码保存在 timeOut2.go 中,并分为五部分来介绍。这次,超时时间作为命令行参数提供给程序。

timeOut2.go 的第一部分如下:

package main

import (
    "fmt"
    "os"
    "strconv"
    "sync"
    "time"
)

timeOut2.go 的第二段代码如下:

func timeout(w *sync.WaitGroup, t time.Duration) bool {
    temp := make(chan int)
    go func() {
        time.Sleep(5 * time.Second)
        defer close(temp)
        w.Wait()
    }()

    select {
        case <-temp:
            return false
        case <-time.After(t):
            return true
    }
}

上面的代码里,time.After() 调用使用的时间周期是 timeout() 函数的一个参数,就是说它可变。再次,select 块实现了超时逻辑。另外,w.Wait() 调用使 timeout() 函数无期限的等待一个匹配的 sync.Done() 函数来结束。当 w.Wait() 调用返回,select 表达式的第一个分支就会执行。

timeOut2.go 的第三段代码如下:

func main() {
    arguments := os.Args
    if len(arguments) != 2 {
        fmt.Println("Need a time duration!")
        return
    }

    var w sync.WaitGroup
    w.Add(1)
    t, err := strconv.Atoi(arguments[1])
    if err != nil {
        fmt.Println(err)
        return
    }

timeOut2.go 的第四部分如下:

    duration := time.Duration(int32(t)) * time.Millisecond
    fmt.Printf("Timeout period is %s\n", duration)

    if timeout(&w, duration) {
        fmt.Println("Timed out!")
    } else {
        fmt.Println("OK!")
    }

time.Duration() 函数转换一个整数值给一个之后使用到的 time.Duration 变量。

timeOut2.go 的其余代码如下:

    w.Done()
    if timeout(&w, duration) {
        fmt.Println("Timed out!")
    } else {
        fmt.Println("OK!")
    }
}

一旦 w.Done() 调用执行,之前的 timeout() 函数将返回。然而第二次调用 timeout() 没有要等待的 sync.Done() 语句。

执行 timeOut2.go 产生如下输出:

$go run timeOut2.go 10000
Timeout period is 10s
Timed out!
OK!

执行 timeOut2.go 的超时周期比匿名 goroutine 的 time.Sleep(5 * time.Second) 长。然而,没有必要的 w.Done() 调用,匿名 goroutine 不能返回因为 time.After() 调用先结束了,所以第一个 if 表达式的 timeout() 函数返回 true。第二个 if 表达式,匿名函数不必等待,因为 time.Sleep(5 * time.Second) 将先于 time.After(t) 结束,所以 timeout() 函数返回 false:

$go run timeOut2.go 100
Timeout period is 100ms
Timed out!
Timed out!

然而,第二次执行,这个超时周期太短了,所以这两个 timeout() 的执行都没有足够的时间完成,因此都超时了。

所以,当定义一个超时周期时,确保您选择了一个恰当的值,否则您的结果可能不是您期望的那样。

最后编辑: kuteng  文档更新时间: 2021-03-27 20:14   作者:kuteng