golang context实例

什么是Go语言的context？

在Go语言中，context是一个非常重要的概念。它可以让我们在不同的goroutine间传递请求相关的数据、取消操作以及设置超时和截止日期。本文将介绍context的基本用法，帮助开发者更好地理解和使用它。

如何使用context？

在开始使用context之前，我们需要先引入"context"包：

import "context"

创建一个context对象

为了创建一个context对象，我们可以使用两种方式：

1. 使用context.Background()创建一个空的context对象。
2. 使用context.WithCancel、context.WithDeadline、context.WithTimeout或context.WithValue等函数，根据需求创建一个具有特定属性的context对象。

取消操作

当我们需要取消一个context时，可以调用它的cancel()方法：

ctx, cancel := context.WithCancel(context.Background())
go func() {
    time.Sleep(time.Second)
    cancel()
}()

超时和截止日期

我们可以使用context.WithTimeout和context.WithDeadline函数，给context设置超时和截止日期。

timeout := 5 * time.Second
ctx, cancel := context.WithTimeout(context.Background(), timeout)
defer cancel()

deadline := time.Now().Add(10 * time.Second)
ctx, cancel := context.WithDeadline(context.Background(), deadline)
defer cancel()

传递请求相关的数据

有时候，在一个复杂的业务流程中，我们需要在不同的goroutine间传递请求相关的数据。context提供了一个键值对的方式来传递这些数据。

type key string

const myKey key = "myKey"

func DoSomething(ctx context.Context) {
    ctx = context.WithValue(ctx, myKey, "myValue")
    
    go func() {
        if value := ctx.Value(myKey); value != nil {
            fmt.Println(value)
        }
    }()
}

总结

通过使用context，我们可以更好地控制和管理goroutine之间的交互。它提供了一种优雅的方式来传递请求相关的数据、取消操作以及设置超时和截止日期。希望本文对于理解和使用golang context有所帮助。