小伙伴们对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的相关知识，请关注编程网公众号！
