Go Context上下文的应用

Go的上下文应用

1.背景原理

Go上下文就是context。对于go来说go routine（协程）是一个非常有用的方法。
本来用chan（通道）也可以达到控制的目的，无奈如果开的协程太多的话，控制起来就很麻烦。光命名都头疼。
后来又有了sync.WaitGroup来控制，不过好像也没有那么优雅。没有怎么研究。
context(上下文)的出现，优雅的解决了这个问题。
context的原理：

clipboard.png 2.png

context可以创建父子结构，树形结构，如上图。

2.举例说明

设定一个需求：
需要建立一组协程，协程图如下：

main-->A1(协程)-->A1_1(协程)->A1_1_1(协程)
|->B(协程)->B1(协程)
A1熔断时间:5s; A_1熔断时间:4s; A_1_1熔断时间:3s
B1错误，全部结束

上源码：

package main

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

func main() {
    contextNodeMain()
}

/*
需求概述
主程序的协程
main->A->A_1->A_1_1
   |->B->B1
   主程序的断熔时间为5s;A为4s;A_1为3.5s;A_1_1为3S;
   若B1错误，主程序强制结束
*/
func contextNodeMain() {
    d := time.Now().Add(5000 * time.Millisecond)
    ctx, cancel := context.WithDeadline(context.Background(), d)
    defer func() { cancel() }()
    //B1结束通道
    ch := make(chan bool)
    go A(ctx)
    go B(ctx, ch)

    select {
    case <-ctx.Done(): //全部声明周期燃尽
        log.Println("所有协程关闭") 
    case <-ch: //B1发生错误全部结束
        log.Println("发生错误,其余协程结束")
        return
        //cancel()
    }

}
func B(ctx context.Context, ch chan bool) {
    log.Println("B 开始工作")
    B_1(ch)
}

func B_1(ch chan bool) {
    time.Sleep(time.Second * 10)
    log.Println("B_1 发生错误")
    ch <- false
}

func A(ctx context.Context) {
    ctx_A, cancel := context.WithTimeout(ctx, time.Millisecond*time.Duration(4000)) //A_1执行时间为3.5秒
    log.Println("A 开始工作")
    defer cancel()
    go A_1(ctx_A)
    for {
        select {
        case <-ctx.Done():
            log.Println("A停止工作")
            return
            // case <-time.After(time.Second):
            //  log.Println("A wroking")
        }
    }

}
func A_1(ctx context.Context) {
    ctx_A_1, cancel := context.WithTimeout(ctx, time.Millisecond*time.Duration(3000)) //A_1_1的执行时间为3秒
    log.Println("A_1 开始工作")
    defer cancel()
    go A_1_1(ctx_A_1)
    for {
        select {
        case <-ctx.Done():
            log.Println("A_1停止工作")
            return
        }
    }
}
func A_1_1(ctx context.Context) {
    log.Println("A_1_1 开始工作")
    for {
        select {
        case <-ctx.Done():
            log.Println("A_1_1停止工作")
            return

        }
    }
}

QQ截图20191106100353.png

从结果中我们可以看到
1.10s时，A和B都开始工作。
2.在13秒的时候，A_1_1停止工作，正好是3S。
3.在14s的时候，A_1停止工作，差4s
4.在15S的时候，协程关闭，差5s

也就是按照我们设想的，停止顺序为A_1_1->A_1->A

如果B_1发生错误，我们只需要改一下B，模拟B_1发生错误修改代码如下

func B_1(ch chan bool) {
    time.Sleep(time.Second * 3)
    log.Println("B_1 发生错误")
    ch <- false
}

结果如下：

QQ截图20191106101013.png

如果B_1发生错误，协程全部关闭。