Golang Context 的使用指南_golang context 使用(1),2024年最新最新大厂Golang校招面试经验汇总

54 阅读3分钟

### 3. 创建 Context


在使用 Context 之前,我们需要先创建一个根 Context。一般情况下,我们会使用 `context.Background()` 函数来创建根 Context。根 Context 是所有其他 Context 的父 Context。



ctx := context.Background()


### 4. 添加超时/取消功能


为了防止请求耗时过长或无法正常响应,我们可以使用 Context 提供的超时功能。通过给 Context 设置一个超时时间,当超过指定的时间后,Context 会自动取消。



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


上述代码将创建一个新的 Context,该 Context 在 5 秒钟之后自动取消。同时,我们也创建了一个 `cancel` 函数,用于在函数退出时显式地取消 Context。


### 5. 传递 Context


有了 Context 后,我们可以将其传递给其他函数,以便它们在需要时使用。现在,让我们看看如何在函数间传递 Context。


#### 5.1 手动传递


在函数的参数列表中,添加一个类型为 `context.Context` 的参数,以便接收调用方传递的 Context。



func Foo(ctx context.Context) { // 在这里使用 ctx }

func Bar(ctx context.Context) { Foo(ctx) }


通过这种方式,我们可以在调用 `Bar` 函数时,将上层函数传递的 Context 传递给 `Foo` 函数,从而实现 Context 在函数间的传递。


#### 5.2 Context WithValue


Context 还提供了 `WithValue` 方法,用于向 Context 中添加一个键值对。这个键值对可以是任意类型的数据,用于传递一些全局信息。



type key int

const ( userIDKey key = iota requestIDKey )

func Foo(ctx context.Context) { userID := ctx.Value(userIDKey).(int) requestID := ctx.Value(requestIDKey).(string)

// 使用 userID 和 requestID

}

func Bar(ctx context.Context, userID int, requestID string) { ctx = context.WithValue(ctx, userIDKey, userID) ctx = context.WithValue(ctx, requestIDKey, requestID)

Foo(ctx)

}


在上述代码中,我们首先定义了两个 Context 的键值(`userIDKey``requestIDKey`)。在 `Bar` 函数中,我们使用 `WithValue` 方法将 `userID``requestID` 添加到 Context 中,然后在 `Foo` 函数中使用这些值。


### 6. 取消 Context


当我们不再需要某个 Context 时,应该显式地取消它,以释放相关资源并停止与该 Context 相关的操作。



func SomeFunc(ctx context.Context) { // 在这里检查 Context 是否已经被取消 select { case <-ctx.Done(): return default: // 继续执行操作 } }


在上述代码中,我们使用 `select` 语句来监听 Context 的 `Done` 通道。一旦 Context 被取消,`Done` 通道就会被关闭,然后我们可以相应地处理取消事件。


### 7. Context 的传播


当一个 Context 被取消时,该取消操作会被传播到该 Context 的所有子 Context。这意味着所有依赖于父 Context 的子 Context 都将被取消。



func Parent(ctx context.Context) { ctx, cancel := context.WithCancel(ctx) defer cancel()

go Child(ctx)

}

img img img

既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,涵盖了95%以上Go语言开发知识点,真正体系化!

由于文件比较多,这里只是将部分目录截图出来,全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频,并且后续会持续更新

如果你需要这些资料,可以戳这里获取