Golang实现context超时控制小demo

go的context上下文机制可以实现go协程的超时控制，以下面代码为例：

package main

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

// 传入context包的Context结构体对象
func LongTimeRunWork(ctx context.Context) {
	for {
		select {
		case <-ctx.Done()://只读队列监听是否收到ctx信号，收到则退出
			fmt.Println("time out")
			return
		default://正常处理逻辑
			fmt.Println("working...")
			time.Sleep(time.Second * 1)
		}
	}
}

func main() {
    // 返回一个函数，用以取消超时控制，使用defer调用是确保主线程推出时能取消
    // context.Background() 是一个空白的上下文，返回一个上下文对象用于go协程的参数作超时控制
	ctx, cancal := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancal()

	go LongTimeRunWork(ctx)

	time.Sleep(10 * time.Second)
}

ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) 这行代码主要用于创建一个具有超时时间限制的上下文 ctx 以及一个用于取消该上下文的函数 cancel 。

context.Background() 是一个默认的空上下文，通常作为创建新上下文的起点。

context.WithTimeout 函数基于给定的父上下文（这里是 context.Background）创建一个新的上下文，并设置了超时时间为 5 秒钟。

ctx 就是新创建的具有超时设置的上下文，在相关的操作中可以通过监听这个上下文的完成信号（比如通过 <-ctx.Done() ）来得知是否已经超时。

cancel 是一个函数，调用它可以主动取消该上下文。在某些情况下，如果在超时之前，您确定不再需要基于这个上下文的操作继续执行，就可以调用 cancel 函数提前取消。

例如，如果在上述代码之后的某个地方，根据一些特定的条件判断需要提前结束相关操作，就可以执行 cancel() 。