首页 >后端开发 >Golang >我们如何保证取消的上下文会导致 goroutine 终止?

我们如何保证取消的上下文会导致 goroutine 终止?

PHPz
PHPz转载
2024-02-08 23:54:23994浏览

我们如何保证取消的上下文会导致 goroutine 终止?

php小编子墨将为大家介绍如何保证取消上下文会导致goroutine终止的方法。当我们在使用goroutine时,有时候需要在某个条件满足时取消它,以避免不必要的计算和资源浪费。为了确保goroutine能够正确终止,我们可以使用context包提供的机制。context包提供了一种在goroutine之间传递请求的方法,并且可以在需要时取消这些请求。通过合理地使用context包,我们可以确保goroutine在取消上下文时能够正确地终止,从而避免资源泄漏和其他潜在的问题。下面我们将详细介绍如何使用context包来实现这一目标。

问题内容

假设发生以下情况:

  • 我们有下面的 consumer 函数,在 goroutine 中运行。

  • 另一个 goroutine 正在 intchan 通道上毫无延迟地发送整数。换句话说,在 for 循环的每次迭代中,intchan 上都有一个准备好接收的值。

  • 启动 consumer goroutine 的 goroutine 已取消传递到 consumer 的上下文。因此,ctx.done() goroutine 的 goroutine 已取消传递到

    的上下文。因此,ctx.done() 通道也有一个可供接收的值。

问题:
  • 在这种情况下,select 语句的两种情况都已准备好运行。
  • select根据 go 之旅,
  • 将随机选择一种情况,因为两者都已准备好运行。
  • select 不会继续选择 <- intchan 案例?如果 <- c​​tx.done() 案例在 for 循环的每次迭代中都准备就绪,我们如何知道 <- c​​tx.done()如何保证
  • 不会继续选择 <- intchan 案例?如果 <- c​​tx.done() 案例在 for 循环的每次迭代中都准备就绪,我们如何知道 <- c​​tx.done() 案例最终会被选择?

func consumer(ctx context.context, intchan chan int) {
    for {
        select {
        case <-ctx.done():
            return
        case i := <-intchan:
            foo(i)
        }
    }
}
consumer我尝试在下面的程序中使用 consumerproducer 函数。 在该程序的多次运行中,

producer goroutine 似乎总是终止。

<-ctx.done()为什么我们最终不会以

案例从未执行的运行结束?
package main

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

func main() {

    ctx, cancelFunc := context.WithCancel(context.Background())

    var wg sync.WaitGroup
    wg.Add(2) // add 2, because we spawn 2 goroutines

    Producer(ctx, &wg)

    fmt.Println(time.Now())
    time.Sleep(time.Second * 5) // cancel the context after 5 seconds

    cancelFunc()
    fmt.Println("CANCELLED")

    wg.Wait() // wait till both producer and consumer goroutines terminate
    fmt.Println(time.Now())

}

func Producer(ctx context.Context, wg *sync.WaitGroup) {
    intChan := make(chan int)

    go Consumer(ctx, intChan, wg)

    go func() {
        defer wg.Done()
        for {
            select {
            case <-ctx.Done():
                return
            case intChan <- 1:
            }
        }
    }()

}

func Consumer(ctx context.Context, intChan chan int, wg *sync.WaitGroup) {
    defer wg.Done()

    for {
        select {
        case <-ctx.Done():
            return
        case _ = <-intChan:
        }
    }
}

解决方法

ctx.err()没有任何保证。保证终止的最简单方法是在 select 语句之外使用

检查错误。将错误返回到传递上下文的代码也很常见。我会这样编写 consumer 函数:🎜
func Consumer(ctx context.Context, intChan chan int) error {
    for ctx.Err() == nil {
        select {
        case <-ctx.Done():
        case i := <-intChan:
            foo(i)
        }
    }
    return ctx.Err()
}

以上是我们如何保证取消的上下文会导致 goroutine 终止?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文转载于:stackoverflow.com。如有侵权,请联系admin@php.cn删除