10.3.2 方式2
第二个技巧的源码保存在 timeOut2.go 中,并分为五部分来介绍。这次,超时时间作为命令行参数提供给程序。
timeOut2.go 的第一部分如下:
1
package main
2
3
import (
4
"fmt"
5
"os"
6
"strconv"
7
"sync"
8
"time"
9
)
Copied!
timeOut2.go 的第二段代码如下:
1
func timeout(w *sync.WaitGroup, t time.Duration) bool {
2
temp := make(chan int)
3
go func() {
4
time.Sleep(5 * time.Second)
5
defer close(temp)
6
w.Wait()
7
}()
8
9
select {
10
case <-temp:
11
return false
12
case <-time.After(t):
13
return true
14
}
15
}
Copied!
上面的代码里,time.After() 调用使用的时间周期是 timeout() 函数的一个参数,就是说它可变。再次,select 块实现了超时逻辑。另外,w.Wait() 调用使 timeout() 函数无期限的等待一个匹配的 sync.Done() 函数来结束。当 w.Wait() 调用返回,select 表达式的第一个分支就会执行。
timeOut2.go 的第三段代码如下:
1
func main() {
2
arguments := os.Args
3
if len(arguments) != 2 {
4
fmt.Println("Need a time duration!")
5
return
6
}
7
8
var w sync.WaitGroup
9
w.Add(1)
10
t, err := strconv.Atoi(arguments[1])
11
if err != nil {
12
fmt.Println(err)
13
return
14
}
Copied!
timeOut2.go 的第四部分如下:
1
duration := time.Duration(int32(t)) * time.Millisecond
2
fmt.Printf("Timeout period is %s\n", duration)
3
4
if timeout(&w, duration) {
5
fmt.Println("Timed out!")
6
} else {
7
fmt.Println("OK!")
8
}
Copied!
time.Duration() 函数转换一个整数值给一个之后使用到的 time.Duration 变量。
timeOut2.go 的其余代码如下:
1
w.Done()
2
if timeout(&w, duration) {
3
fmt.Println("Timed out!")
4
} else {
5
fmt.Println("OK!")
6
}
7
}
Copied!
一旦 w.Done() 调用执行,之前的 timeout() 函数将返回。然而第二次调用 timeout() 没有要等待的 sync.Done() 语句。
执行 timeOut2.go 产生如下输出:
1
$go run timeOut2.go 10000
2
Timeout period is 10s
3
Timed out!
4
OK!
Copied!
执行 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:
1
$go run timeOut2.go 100
2
Timeout period is 100ms
3
Timed out!
4
Timed out!
Copied!
然而,第二次执行,这个超时周期太短了,所以这两个 timeout() 的执行都没有足够的时间完成,因此都超时了。
所以,当定义一个超时周期时,确保您选择了一个恰当的值,否则您的结果可能不是您期望的那样。
Last modified 2yr ago
Copy link