golang context学习记录2

上篇文章中，我们已经学习了使用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...