golang context学习记录1

ゞ浴缸里的玫瑰 2022-01-30 13:15 259阅读 0赞

# 1.前言 #

一个请求，可能涉及多个API调用，多个goroutine，如何在多个API 之间，以及多个goroutine之间协作和传递信息，就是一个问题。

比如一个网络请求Request，需要开启一些goroutine去访问后端资源(比如，数据库，RPC服务等)，这些goroutine又可能会开启其他的goroutine，如何跟踪和控制这些goroutine呢？

golang定义了 context包，用于解决这个问题。

`context`可以在多个API和进程之间，实现`deadlines`截止日期操作、`cancelation signals`取消操作以及传递`request-scoped`的值(即传递参数)。

`context`一个最大的特点是可以继承。父context，可以派生子context，子context有又可以派生子context。如果父context被取消，子context及其子context都会被取消。

常用派生的方法有:`WithCancel` `WithDeadline` `WithTimeout` `WithValue`.  
其中，

*  `WithCancel`用于主动取消自身以及子context。
 *  `WithDeadline` 和`WithTimeout`用于超时情况下的取消自身以及子context。`WithTimeout`实现上直接调用`WithDeadline`。
 *  `WithValue`用于`context`传递`request-scoped`的参数，而不是向函数传递的可选参数。

# 2.例子 #

下面例子演示如何控制多个goroutine的退出。

首先，启动3个goroutine，分别命名为"1"、“2”、“3”，每个goroutine又启动一个goroutine。也就是有6个goroutine。它们之间的关系如下：

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

最后，在`main`中调用取消操作`cancel`，先是1，2，3收到信号退出，1，2，3退出的时候又调用它们的`cancel`，这样所有的goroutine都被取消并退出。

具体代码如下：

package main
    
    import (
            "context"
            "fmt"
            "time"
    )
    
    
    func PrintTask(ctx context.Context, taskName string) { 
    
            for { 
    
                    select { 
    
                    case <- ctx.Done():
                            fmt.Println("task:", taskName, " 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, " exit...")
                            return
                    default:
                            time.Sleep(1*time.Second)
                            fmt.Println("task:", taskName, " doing something...")
                    }
    
            }
    
    
    }
    
    func main() { 
    
            ctx := context.Background()
            ctx, cancel := context.WithCancel(ctx)
    
            go mainTask(ctx, "1")
            go mainTask(ctx, "2")
            go mainTask(ctx, "3")
    
            time.Sleep(3*time.Second)
            cancel()
            fmt.Println("main exit...")
            time.Sleep(3*time.Second)
    }

输出

task: 1  doing something...
    task: 21  doing something...
    task: 11  doing something...
    task: 3  doing something...
    task: 2  doing something...
    task: 31  doing something...
    task: 3  doing something...
    task: 2  doing something...
    task: 11  doing something...
    task: 21  doing something...
    task: 31  doing something...
    task: 1  doing something...
    task: 11  doing something...
    task: 3  doing something...
    task: 1  doing something...
    task: 21  doing something...
    task: 2  doing something...
    task: 31  doing something...
    main exit...
    task: 11  doing something...
    task: 11  exit...
    task: 21  doing something...
    task: 21  exit...
    task: 3  doing something...
    task: 3  exit...
    task: 31  doing something...
    task: 31  exit...
    task: 1  doing something...
    task: 1  exit...
    task: 2  doing something...
    task: 2  exit...

# 更多参考 #

[Golang并发控制–context的使用][Golang_context]

[Golang Context深入理解][Golang Context]

[Golang_context]: https://blog.csdn.net/lanyang123456/article/details/80730923
[Golang Context]: https://juejin.im/post/5a6873fef265da3e317e55b6