结束长时间执行的函数：如何在超时后停止

IT行业相对于一般传统行业，发展更新速度更快，一旦停止了学习，很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习，精进自己的技术，尤其是初学者。今天golang学习网给大家整理了《结束长时间执行的函数：如何在超时后停止》，聊聊，我们一起来看看吧！

如果某些事情花费太长时间，我会尝试关闭长时间运行的功能，也许只是治标不治本的解决方案，但无论如何，对于我的情况来说，它并没有真正解决问题。

我是这样做的：

func foo(abort <- chan struct{}) {
for {
  select{
    case <-abort:
      return
    default:
    ///long running code
  }
}
}

在单独的函数中，我有一段时间后会关闭传递的链，如果我剪切主体返回该函数，它就会这样做。但是，如果有一些长时间运行的代码，它不会影响结果，它只是继续工作，就好像什么也没发生一样。

感觉应该可以，但实际上不行。我有什么遗漏的吗？毕竟所有的路由器框架都有超时功能，之后任何正在运行的东西都会被终止。所以也许这只是出于好奇，但我真的很想知道如何做到这一点。

正确答案

在“长时间运行的代码”中，您必须定期检查 abort 通道。

“定期”实现这一点的通常方法是将代码分成多个块，每个块在相当短的时间内完成（假设进程运行的系统没有过载）。

执行完每个这样的块后，检查终止条件是否成立，如果成立则终止执行。
执行此类检查的惯用方法是“select with default”：

select {
case <-channel:
  // terminate processing
default:
}

这里，如果 channel 尚未准备好接收（或关闭），则立即采用 default 无操作分支。

一些算法使这种分块变得更容易，因为它们采用了一个循环，其中每次迭代执行的时间大致相同。 如果你的算法不是这样的，你就必须手动对其进行分块；在这种情况下，最好为每个块创建一个单独的函数（或方法）。

更多要点。

1. 考虑使用 contexts：它们提供了一个有用的框架来解决与您正在解决的问题类似的问题。

   更好的是，它们可以相互“继承”，这一事实使人们能够轻松实现两个巧妙的事情：

   • 您可以组合各种方法来取消上下文：例如，可以创建一个上下文，该上下文可以在超时后取消，也可以通过其他代码显式取消。
   • 它们使得创建“取消树”成为可能——当取消根上下文时，将此信号传播到所有继承上下文——使它们取消其他 goroutine 正在执行的操作。
有时，当人们说“长时间运行的代码”时，他们并不是指实际上一直在 cpu 上处理数字的代码，而是指对慢速实体（例如数据库、http 服务器等）执行请求的代码, — 在这种情况下，代码实际上并未运行，而是在 i/o 上休眠以传递一些要处理的数据。

如果您的情况如此，请注意所有编写良好的 go 包（当然，这包括 go 标准库中处理网络服务的所有包）都接受其 api 的那些函数中的上下文，这些函数实际上调用如此缓慢的实体，这意味着如果您让您的函数接受上下文，您可以（实际上应该）将此上下文传递到适用的调用堆栈中 - 以便您调用的所有代码都可以以与以下相同的方式取消你的。

进一步阅读：

• https://go.dev/blog/pipelines
• https://blog.golang.org/advanced-go-concurrency-patterns

在执行长时间运行的代码之前，您的代码仅在每次迭代时检查通道是否关闭一次。在长时间运行的代码开始后，没有机会检查 abort chan，因此它将运行到完成。

您需要偶尔检查是否在长时间运行的代码主体中提前退出，使用 context.context 和 withtimeout 可以更惯用地完成此操作，例如：https://pkg.go.dev/context#example-WithTimeout

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~