温馨提示×

在Go语言中如何处理并发任务取消问题

小云
110
2023-10-09 08:21:49
栏目: 编程语言
GO开发者专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

在Go语言中处理并发任务取消问题,可以使用context包来管理和取消任务。

首先,使用context包创建一个上下文对象,例如:

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

然后,在并发的任务中通过select语句来监听取消信号,例如:

select {
case <-ctx.Done():
// 任务被取消,执行清理操作
return
default:
// 执行任务操作
// ...
}

如果要取消任务,可以调用cancel函数,例如:

cancel()

当调用cancel函数时,所有使用同一个context对象的协程都会收到取消信号,从而停止执行任务。

完整的示例代码如下:

package main
import (
"context"
"fmt"
"time"
)
func main() {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
go worker(ctx, "worker1")
go worker(ctx, "worker2")
time.Sleep(3 * time.Second)
cancel()
time.Sleep(1 * time.Second)
}
func worker(ctx context.Context, name string) {
for {
select {
case <-ctx.Done():
fmt.Printf("%s: 任务被取消\n", name)
return
default:
fmt.Printf("%s: 执行任务\n", name)
time.Sleep(1 * time.Second)
}
}
}

运行上述代码,会输出如下结果:

worker1: 执行任务
worker2: 执行任务
worker1: 执行任务
worker2: 执行任务
worker1: 任务被取消
worker2: 任务被取消

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:Go语言中如何处理并发任务重试问题

0