使用Go语言中的Context取消协程执行的操作代码_Golang

使用 go 语言中的 context 取消协程执行

在 go 语言中，协程（goroutine）是一种轻量级的线程，非常适合处理并发任务。然而，如何优雅地取消正在运行的协程是一个常见的问题。本文将通过一个具体的例子来展示如何使用 context 包来取消协程的执行，特别是处理嵌套任务中的取消问题。

问题描述

假设我们有一个长时间运行的任务，该任务包含一个外层循环和一个内层任务。我们需要在外层循环接收到取消信号时，能够立即终止内层任务。以下是一个示例代码：

package main

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

// longrunningtask 是一个模拟长时间运行的任务。
func longrunningtask(ctx context.context) {
	for {
		select {
		case <-ctx.done(): // 监听 ctx.done() 以获取取消信号
			fmt.println("任务被取消:", ctx.err())
			return // 接收到取消信号后退出
		default:
			currenttime := time.now().format("2006-01-02 15:04:05") // 获取并格式化当前时间
			fmt.printf("任务进行中... 当前时间：%s\n", currenttime)
			for {
				fmt.printf("111")
				time.sleep(1 * time.second) //
			}
		}
	}
}

func main() {
	// 创建一个可以取消的 context
	ctx, cancel := context.withcancel(context.background())

	// 启动一个新的 goroutine 执行任务
	go longrunningtask(ctx)

	// 模拟一段时间后取消任务
	time.sleep(3 * time.second)
	fmt.println("取消任务...")
	cancel() // 发送取消信号

	// 等待一段时间让任务有时间处理取消信号并退出
	time.sleep(10 * time.second)
}

在这个示例中，当我们取消任务时，外层循环会接收到取消信号并退出，但内层循环会继续运行，因为我们没有在内层循环中检查取消信号。

解决方案

为了确保内层任务也能响应取消信号，我们需要在内层任务中也检查 ctx.done() 通道。以下是修改后的代码：

package main

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

// longrunningtask 是一个模拟长时间运行的任务。
func longrunningtask(ctx context.context) {
	for {
		select {
		case <-ctx.done(): // 监听 ctx.done() 以获取取消信号
			fmt.println("任务被取消:", ctx.err())
			return // 接收到取消信号后退出
		default:
			currenttime := time.now().format("2006-01-02 15:04:05") // 获取并格式化当前时间
			fmt.printf("任务进行中... 当前时间：%s\n", currenttime)
			// 启动内层任务
			runinnertask(ctx)
		}
	}
}

// runinnertask 是一个模拟内层长时间运行的任务。
func runinnertask(ctx context.context) {
	for {
		select {
		case <-ctx.done(): // 内层任务也监听 ctx.done()
			fmt.println("内层任务被取消:", ctx.err())
			return // 接收到取消信号后退出
		default:
			fmt.printf("111")
			time.sleep(1 * time.second)
		}
	}
}

func main() {
	// 创建一个可以取消的 context
	ctx, cancel := context.withcancel(context.background())

	// 启动一个新的 goroutine 执行任务
	go longrunningtask(ctx)

	// 模拟一段时间后取消任务
	time.sleep(3 * time.second)
	fmt.println("取消任务...")
	cancel() // 发送取消信号

	// 等待一段时间让任务有时间处理取消信号并退出
	time.sleep(10 * time.second)
}