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

 更新时间:2024年11月29日 11:08:23   作者:软件架构师笔记  
在 Go 语言中,协程(goroutine)是一种轻量级的线程,非常适合处理并发任务,然而,如何优雅地取消正在运行的协程是一个常见的问题,本文将通过一个具体的例子来展示如何使用 context 包来取消协程的执行,需要的朋友可以参考下

使用 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)
}

解释

外层循环:

外层循环使用 select 语句来监听 ctx.Done() 通道。如果接收到取消信号,任务会打印一条消息并退出。

内层任务:

内层任务也使用 select 语句来监听 ctx.Done() 通道。如果接收到取消信号,内层任务会打印一条消息并退出。

通过这种方式,我们可以确保无论是在外层循环还是内层任务中,任务都能响应取消信号并优雅地退出。

总结

在 Go 语言中,使用 context 包来管理协程的生命周期是非常重要的。通过在每个需要响应取消信号的地方检查 ctx.Done() 通道,我们可以确保任务能够及时响应取消信号并优雅地退出。这对于构建健壮和可靠的并发应用程序至关重要。

到此这篇关于使用Go语言中的Context取消协程执行的操作代码的文章就介绍到这了,更多相关Go Context取消协程执行内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Golang标准库之errors包应用方式

    Golang标准库之errors包应用方式

    Go语言的errors包提供了基础的错误处理能力,允许通过errors.New创建自定义error对象,error在Go中是一个接口,通过实现Error方法来定义错误文本,对错误的比较通常基于对象地址,而非文本内容,因此即使两个错误文本相同
    2024-10-10
  • 一文详解Golang中net/http包的实现原理

    一文详解Golang中net/http包的实现原理

    这篇文章主要介绍了如何用 net/http 自己编写实现一个 HTTP Server 并探究其实现原理,具体讲解Go语言是如何接收和处理请求的,希望能够对大家的学习或工作具有一定的帮助
    2022-08-08
  • 示例剖析golang中的CSP并发模型

    示例剖析golang中的CSP并发模型

    这篇文章主要为大家介绍了示例剖析golang中的CSP并发模型,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-05-05
  • golang gorm学习之如何指定数据表

    golang gorm学习之如何指定数据表

    在sql中首先要指定是从哪张表中查询,所以这篇文章小编就来带大家一起看一下gorm是如何根据model来自动解析表名的,感兴趣的小伙伴可以了解下
    2023-08-08
  • golang gorm的关系关联实现示例

    golang gorm的关系关联实现示例

    这篇文章主要为大家介绍了golang gorm的关系关联实现示例,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步早日升职加薪
    2022-04-04
  • 图文详解go语言反射实现原理

    图文详解go语言反射实现原理

    这篇文章主要介绍了图文详解go语言反射实现原理,本文图文并茂给大家介绍的非常详细,具有一定的参考借鉴价值,需要的朋友参考下吧,需要的朋友可以参考下
    2020-02-02
  • Go语言中的定时器原理与实战应用

    Go语言中的定时器原理与实战应用

    在Go语言中,Timer和Ticker是处理定时任务的重要工具,Timer用于一次性事件,而Ticker则用于周期性事件,本文详细介绍了这两种定时器的创建、使用和停止方法,并通过实际案例展示了它们在监控日志、检查系统状态等方面的应用
    2024-10-10
  • Go使用path/filepath包处理文件路径的完全指南

    Go使用path/filepath包处理文件路径的完全指南

    本文详细介绍了Go语言中path/filepath包的使用,涵盖了路径处理、拼接、规范化、遍历和匹配等功能,适用于文件系统操作的各种场景,需要的朋友可以参考下
    2026-03-03
  • 深入了解Go语言中web框架的中间件运行机制

    深入了解Go语言中web框架的中间件运行机制

    大家在使用iris框架搭建web系统时,一定会用到中间件。那么你了解中间件的运行机制吗?你知道为什么在iris和gin框架的请求处理函数中要加c.Next()函数吗?本文就和大家一起探究该问题的答案
    2023-02-02
  • 解决go 生成的exe不在bin文件夹里的问题

    解决go 生成的exe不在bin文件夹里的问题

    这篇文章主要介绍了解决go 生成的exe不在bin文件夹里的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-12-12

最新评论