Golang中实现无限循环并设置超时

从现在开始，我们要努力学习啦！今天我给大家带来《Golang中实现无限循环并设置超时》，感兴趣的朋友请继续看下去吧！下文中的内容我们主要会涉及到等等知识点，如果在阅读本文过程中有遇到不清楚的地方，欢迎留言呀！我们一起讨论，一起学习！

我正在尝试读取恒定的数据流，如果接收流的调用时间超过 30 秒，我需要超时并退出程序。我不知道如何在收到超时后退出 go 例程。

func ReceiveStreamMessages(strm Stream, msg chan<- []byte) error {
  d := make(chan []byte, 1)

  e := make(chan error)

  tm := time.After(30 * time.Second)

  go func() {
    for {
        //blocking call
        data, err := strm.Recv()
        if err != nil {
            e <- err
            return
        }
        select {
        case d <- data.Result:
        case <-tm:
            //exit out go routine
            return
         }
      }
  }()

  for {
    select {
    case message := <-d:
        msg <- message
    case err := <-e:
        return err
    case <-tm:
        return nil
    }
  }
}

我上面的代码是错误的：为了让 select 在 go 例程 for 循环中运行，阻塞函数必须返回并且数据将被填充，因此不会遇到超时选择情况（或者会做随机，因为两者都准备好了）。退出父函数是否足以退出 go 例程？

解决方案

使用context包withtimeout。像这样的事情：

package main

import (
    "context"
    "fmt"
    "sync"
    "time"
)

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
    // prepare
    ...
    // wait group just for test
    var wg sync.WaitGroup
    wg.Add(1)
    go func() {
        for {
            select {
            case d <- data.Result:
               // do something
            case <-ctx.Done():
                fmt.Println("Done")
                wg.Done()
                return
            }
        }
    }()
    wg.Wait()
    cancel()
    fmt.Println("Hello, playground")
}

您可以在此处查看工作示例 https://play.golang.org/p/agi1fimtEkJ

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Golang中实现无限循环并设置超时》文章吧，也可关注golang学习网公众号了解相关技术文章。