goroutine 没有看到上下文取消？

小伙伴们对Golang编程感兴趣吗？是否正在学习相关知识点？如果是，那么本文《goroutine 没有看到上下文取消？》，就很适合你，本篇文章讲解的知识点主要包括。在之后的文章中也会多多分享相关知识点，希望对大家的知识积累有所帮助！

我有两个 goroutine 同时运行。

在某些时候，我希望我的程序能够正常退出，因此我使用 cancel() 函数来通知我的 goroutine 它们需要停止，但两者中只有一个收到消息。

这是我的主要内容（简化）：

ctx := context.background()
ctx, cancel := context.withcancel(ctx)

done := make(chan os.signal, 1)
signal.notify(done, os.interrupt, syscall.sigint, syscall.sigterm)

wg := &sync.waitgroup{}
wg.add(2)

go func() {
    err := eng.watcher(ctx, wg)
    if err != nil {
        cancel()
    }
}()

go func() {
    err := eng.suspender(ctx, wg)
    if err != nil {
        cancel()
    }
}()

<-done // wait for sigint / sigterm
log.print("receive shutdown")
cancel()
wg.wait()

log.print("controller exited properly")

suspender goroutine 成功存在（代码如下）：

package main

import (
    "context"
    "sync"
    "time"

    log "github.com/sirupsen/logrus"
    metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
    "k8s.io/client-go/util/retry"
)

func (eng *engine) suspender(ctx context.context, wg *sync.waitgroup) error {

    contextlogger := eng.logger.withfields(log.fields{
        "go-routine": "suspender",
    })
    contextlogger.info("starting suspender goroutine")
    now := time.now().in(eng.loc)

    for {
        select {
        case n := <-eng.wl:
            //dostuff


        case <-ctx.done():
            // the context is over, stop processing results
            contextlogger.infof("goroutine suspender canceled by context")
            return nil
        }
    }

}

这是未接收上下文取消的函数：

package main

import (
    "context"
    "sync"
    "time"

    log "github.com/sirupsen/logrus"
)

func (eng *Engine) Watcher(ctx context.Context, wg *sync.WaitGroup) error {
    contextLogger := eng.logger.WithFields(log.Fields{
        "go-routine":      "Watcher",
        "uptime-schedule": eng.upTimeSchedule,
    })
    contextLogger.Info("starting Watcher goroutine")

    ticker := time.NewTicker(time.Second * 30)
    for {
        select {
        case <-ctx.Done():
            contextLogger.Infof("goroutine watcher canceled by context")
            log.Printf("toto")
            return nil
        case <-ticker.C:
            
                //dostuff
            }
        }
    }
}

你能帮我吗？

谢谢:)

正确答案

你用 errgroup 尝试过吗？它内置了上下文取消功能：

ctx := context.Background()
ctx, cancel := context.WithCancel(ctx)
defer cancel()

done := make(chan os.Signal, 1)
signal.Notify(done, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)

// "golang.org/x/sync/errgroup"
wg, ctx := errgroup.WithContext(ctx)

wg.Go(func() error {
    return eng.Watcher(ctx, wg)
})

wg.Go(func() error {
    return eng.Suspender(ctx, wg)
})

wg.Go(func() error {
    defer cancel()
    <-done
    return nil
})

err := wg.Wait()
if err != nil {
    log.Print(err)
}

log.Print("receive shutdown")
log.Print("controller exited properly")

本篇关于《goroutine 没有看到上下文取消？》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！