Golang Go语言中怎么在主程简单对函数进行超时控制

协程的超时控制可以用 select 简单实现，请问如果是在 for 循环里，每次循环执行一个函数，要求对这个函数进行超时控制(超时就直接退出函数)，有什么优雅的实现方法?

Golang Go语言中怎么在主程简单对函数进行超时控制
20 回复

select 已经算是优雅的了

更多关于Golang Go语言中怎么在主程简单对函数进行超时控制的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


不好意思，我这边可能说得比较糊，就是主程序里有个函数 RunTask()在运行，怎么对这个函数进行超时控制呢，这个函数不是在协程中运行的

想要”控制”，必然要和 main 分开运行，那不就是要用协程嘛

或者函数主动实现超时退出，也是要用协程，不过逻辑在函数里面。

那么这个函数里面必须有 for 循环，每次循环的时候判断一下。其实就是这个函数要是可分的、不一定是循环。

比如
func a(){ time.Sleep(time.Hour) }
就不行，而
func b(){
time.Sleep(time.Minute)
time.Sleep(time.Minute)
time.Sleep(time.Minute)
}
就可以，超时时间精度是 1 分钟

我最近也遇到类似的问题了，不是协程。我是定时任务，执行一个任务执行比较花时间。应该是执行了一部分后，可能是超时退出了，但是不知道怎么控制这个超时时间。导致后面的任务没执行。

context

前段时间讨论过了 /t/851321
答案是没有办法从外部强制终止，只能让协程自己退出，比较好的方式是用 context 控制

context.WithTimeout

将 RunTask 转成一个 goroutine ，然后通过 select 以及 context 做控制。

没办法, 如果是进程 /线程可以通过定时器+kill/tgkill 实现, go 这种自己实现的协程就没有特别好的办法了.

嗯嗯，看来也只能这样了

我想实现的是对 mysql 进行连接测试，发现这个标准库并没有超时控制，而 ssh 的连接测试是有自带一个 timeout 参数的。不过看了 net.DialTimeout 函数发现这个也是用多开一个 go 程+select 实现超时控制的。看来也只能用 go 程了。

用装饰器模式包装函数

go<br>type RunTask func(int) float64<br><br>func RunTaskTimeout(fun RunTask, timeout time.Duration) RunTask {<br><br> return func(i int) float64 {<br><br> ch := make(chan float64)<br><br> go func() {<br> ch &lt;- func(i int) float64 {<br> return func(i int) float64 {<br> return fun(i)<br> }(i)<br> }(i)<br> }()<br><br> select {<br> case f := &lt;-ch:<br> return f<br> case &lt;-time.After(timeout):<br> return -1<br> }<br> }<br>}<br><br>func main() {<br> runTask := RunTaskTimeout(Pi, time.Second)<br> data := []int{500, 5000, 50000, 500000, 5000000}<br> for i := 0; i &lt; 10; i++ {<br> if res := runTask(data[i]); res &lt; 0 {<br> break<br> } else {<br> fmt.Println(res)<br> }<br> }<br> fmt.Println("DONE.")<br>}<br><br>

为什么要拒绝使用 goroutine 呢，配上 context 非常优雅了

学到了，感谢~

#13 感觉这么多闭包没必要呀，直接 ch <- fun(i) 不就可以了

测试 mysql 连接? 用这个库 github.com/go-sql-driver/mysql ?
这个库不是可以带有 timeout 参数吗？

嗯嗯，我后来发现了

我是这么干的，感觉没那么优雅

func main(){
go stopit()
}

func stopit() {
time.AfterFunc(3*time.Second, func() {
log.Println(“stop it”)
})
}

在Golang中，对函数进行超时控制是一个常见的需求，特别是在处理网络请求或长时间运行的任务时。你可以通过以下几种方式来实现超时控制：

  1. 使用context包： Go的context包提供了一种管理跨API和进程间截止日期、取消信号以及其他请求范围的值的方法。你可以创建一个带超时的上下文，并将其传递给需要超时控制的函数。

    示例代码：

    ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()

select {
case <-ctx.Done():
    fmt.Println("Function timed out")
case result := <-longRunningFunction(ctx):
    fmt.Println("Function result:", result)
}

    在longRunningFunction内部，定期检查ctx.Done()是否关闭，如果关闭则提前返回。

  2. 使用goroutine和通道： 另一种方法是启动一个goroutine来运行函数，并使用通道来传递结果或超时信号。不过，这种方法不如使用context包直接和优雅。

  3. 使用第三方库： 一些第三方库提供了更高级的超时控制功能，例如timeout或hystrix等，但通常使用标准库中的context包已经足够满足大多数需求。

总之，使用context包是实现函数超时控制的最推荐方式，它不仅简洁而且功能强大，能够很好地处理各种超时和取消的场景。

回到顶部