golang context学习记录2

青旅半醒 2022-01-30 13:21 136阅读 0赞

[上篇文章][Link 1]中，我们已经学习了使用context实现控制多个goroutine的退出。  
本文将继续介绍如何使用context实现超时情况下，让多个goroutine退出。

# 例子 #

首先，启动3个goroutine，分别为1,2,3。这3个goroutine又分别启动一个goroutine。

一共有6个goroutine，它们之间关系如下：

*  1
    
     *  11
 *  2
    
     *  21
 *  3
    
     *  31

task1,2,3被限制在指定时间内结束，即50ms。

如果不能按时结束，会触发超时操作，这样，task 1、2、3接收到超时信号，取消自己启动的goroutine(即task11,21,31)，然后退出。  
最后，所有的goroutine都退出了。

代码如下：

ackage main
    
    import (
            "context"
            "fmt"
            "time"
    )
    
    
    func PrintTask(ctx context.Context, taskName string) { 
    
            for { 
    
                    select { 
    
                    case <- ctx.Done():
                            fmt.Println("task:", taskName, "ctx error:", ctx.Err()," then task exit...")
                            return
                    default:
                            time.Sleep(1*time.Second)
                            fmt.Println("task:", taskName, " doing something...")
                    }
    
            }
    
    }
    
    func mainTask(ctx context.Context, taskName string) { 
    
            ctx1, cancel := context.WithCancel(ctx)
            defer cancel()
    
            // create a new task
            newTaskName := taskName + "1"
            go PrintTask(ctx1, newTaskName)
    
    
            for { 
    
                    select { 
    
                    case <- ctx.Done():
                            fmt.Println("task:", taskName, "ctx error:", ctx.Err()," then task exit...")
                            return
                    default:
                            time.Sleep(1*time.Second)
                            fmt.Println("task:", taskName, " doing something...")
                    }
    
            }
    
    
    }
    
    func main() { 
    
            ctx := context.Background()
            timeout := 50*time.Millisecond
            ctx, _ = context.WithTimeout(ctx, timeout)
    
            go mainTask(ctx, "1")
            go mainTask(ctx, "2")
            go mainTask(ctx, "3")
    
            select{ 
            case <-ctx.Done():
                    // timeout
                    fmt.Println("main task error:", ctx.Err())
            }
    
            fmt.Println("main exit...")
            time.Sleep(3*time.Second)
    }

输出

main task error: context deadline exceeded
    main exit...
    task: 21  doing something...
    task: 1  doing something...
    task: 21 ctx error: context deadline exceeded  then task exit...
    task: 3  doing something...
    task: 3 ctx error: context deadline exceeded  then task exit...
    task: 1 ctx error: context deadline exceeded  then task exit...
    task: 11  doing something...
    task: 11 ctx error: context deadline exceeded  then task exit...
    task: 31  doing something...
    task: 2  doing something...
    task: 31 ctx error: context deadline exceeded  then task exit...
    task: 2 ctx error: context deadline exceeded  then task exit...

[Link 1]: https://blog.csdn.net/lanyang123456/article/details/90345484