在 Go 中如何使用超时和重试来管理上下文？

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《在 Go 中如何使用超时和重试来管理上下文？》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

我尝试在 go 中使用超时和多次重试来创建上下文。 这是代码示例

func readretry(port io.readwritecloser, timeout, cnt int) []byte {
    fmt.println("in read retry")
    for i := 0; i < cnt; i++ {
        fmt.println("read attempt:", i)
        res := readwithcontext(timeout, port)
        if res != nil {
            return res
        }
    }
    return nil
}

func readwithcontext(timeout int, port io.readwritecloser) []byte {
    fmt.println("in readwithcontext")
    fmt.println("opening channel")
    rcvch := make(chan []byte)
    ctx, cancel := context.withtimeout(context.background(), time.duration(time.second*time.duration(timeout)))
    defer cancel()

    go reader(ctx, port, rcvch)

    for {
        select {
        case <-ctx.done():
            fmt.println("reader: context cancelled")
            return nil
        case buf := <-rcvch:
            fmt.println("reader: got data")
            return buf
        }
    }
}

func reader(ctx context.context, port io.readwritecloser, rcvch chan []byte) {

    fmt.println("in reader")

    answ := make([]byte, 1024)
    buf := bytes.buffer{}
    var err error

    for {
        i := 0
        i, err = port.read(answ)
        if err != nil && err != io.eof {
            log.printf("port.read: %v", err)
        }
        if i != 0 {
            answ = answ[:i]
            buf.write(answ)
            if buf.bytes()[len(buf.bytes())-1] == delimiter {
                fmt.print("received: ")
                printbuf(buf.bytes())
                rcvch <- buf.bytes() //if there is no data in the first attempt, cannot write to the channel here!!
                return
            }
        }
    }
}

然后，我调用 readretry result := readretry(port, 2, 5) // 2 秒超时，5 次重试。但如果第一次数据没有准备好，那么 reader 无法写入 rcvch 。可能已经满了？为什么？如果我尝试在 readwithcontext 执行结束时关闭通道，则会发生冲突 - 写入已关闭的通道。碰撞在哪里？它认为，readwithcontext每次都会作为一个新实例启动，创建一个rcvch的新实例，如果reader因超时而关闭，则所有链函数及其局部变量（包括通道）已被破坏。但是，看来我犯了一个错误。那么，如何进行重试呢？ 看看日志是什么样子的：

IN READ RETRY
Read attempt: 1
IN readWithContext
Opening channel
IN reader
Start reader
Received: 0x01 0x02 0x03 0x04 0x05 0x06 0x07 0x08 0x09 0x0d //<- data is received, but not written to the channel!!
reader: context cancelled

正确答案

每次重试都会创建一个新的阅读器和一个新的频道。如果 readWithContext 超时，读者仍然在那里等待，并且可能最终阅读，但现在通道的另一端没有人在监听，因此读者被泄露。

有一个 reader goroutine 和一个通道，使用 readWithContext 从中读取。如果上下文过期并且所有重试都用尽，您也必须停止读取器。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《在 Go 中如何使用超时和重试来管理上下文？》文章吧，也可关注golang学习网公众号了解相关技术文章。