Golang Context 的并发安全性探究_golang context并发安全

虽然 Context 在设计上是并发安全的,但在实际使用中,我们仍然需要遵循一些最佳实践,以确保代码的正确性和可维护性。

2.1. 避免 Context 滥用

一种常见的误用 Context 的方式是在函数之间传递大而全的 Context,其中包含了过多的信息。这样做不仅增加了代码的复杂度,还可能导致上下文信息的混乱和不一致。

最好的做法是根据具体的需求,只在需要的时候传递相关的 Context 信息。将 Context 限制在必要的范围内,只传递需要的上下文信息,可以使代码更加清晰和可维护。

2.2. Context 值的传递

在多个 goroutine 之间传递 Context 时,确保传递的是 Context 的值,而不是指针。由于 Context 在并发使用时是不可变的,因此复制一个值传递给其他 goroutine 是安全的。

func Worker(ctx context.Context) {
    // 复制 ctx 的值到本地变量
    localCtx := ctx

    // 使用 localCtx 进行操作
}

通过将 Context 的值复制给本地变量,我们可以避免在操作 Context 时的竞争条件。

2.3. 避免在子 goroutine 中创建和取消 Context

Context 的取消操作会导致所有依赖于该 Context 的子 Context 也被取消。因此,在子 goroutine 中创建和取消 Context 可能会产生意外的结果。

通常情况下,我们应该在主 goroutine 中创建根 Context,并在需要的时候将其传递给子 goroutine。子 goroutine 应该由主 goroutine 负责取消。

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

    go Child(ctx)
}

func Child(ctx context.Context) {
    // 使用 ctx 进行操作
}

2.4. 使用 select 监听 Context 的取消

在使用 Context 进行并发操作时,通常还需要同时监听 Context 的取消事件。通过使用 select 语句,我们可以同时监听多个事件,包括 Context 的取消。

func SomeFunc(ctx context.Context) {
    select {
    case <-ctx.Done():
        // Context 取消后的处理逻辑
    case <-otherEventChan:
        // 其他事件的处理逻辑
    }
}

通过使用 select 语句,我们可以及时响应 Context 的取消事件,并执行相应的清理操作。

3. 案例

当使用 Golang Context 进行并发编程时,以下是三个示例案例,展示了如何正确使用 Context 并保证其在并发环境下的安全性。

1. HTTP 请求超时处理

在进行 HTTP 请求时,我们经常需要设置超时时间并在超时时取消请求。使用 Context 可以很方便地实现这一点。

func HTTPRequestWithTimeout(ctx context.Context, url string) error {
    client := http.Client{}

    req, err := http.NewRequest("GET", url, nil)
    if err != nil {
        return err
    }

    // 设置超时时间
    timeoutCtx, cancel := context.WithTimeout(ctx, 5\*time.Second)
    defer cancel()
    req = req.WithContext(timeoutCtx)

    // 发起 HTTP 请求
    resp, err := client.Do(req)
    if err != nil {
        return err
    }
    defer resp.Body.Close()

    // 处理响应
    // ...

    return nil
}

通过使用 WithTimeout 函数创建一个带有超时时间的子 Context,我们可以实现在超时时取消 HTTP 请求。这样可以避免请求等待过长时间,同时确保并发执行多个请求时的安全性。

2. 数据库事务管理

在并发访问数据库时,使用 Golang Context 可以有效地管理数据库事务,并在发生错误或上下文取消时回滚事务。

func DBTransaction(ctx context.Context, db \*sql.DB) error {
    tx, err := db.Begin()
    if err != nil {
        return err
    }

    // 将数据库连接封装到 Context 中
    txCtx := context.WithValue(ctx, "db", tx)

    // 在协程中运行业务逻辑
    errCh := make(chan error)
    go func() {
        // 使用封装了数据库连接的 Context 进行操作
        err := DoBusinessLogic(txCtx)
        if err != nil {
            errCh <- err
        }
        close(errCh)
    }()


## 学习路线:

这个方向初期比较容易入门一些,掌握一些基本技术,拿起各种现成的工具就可以开黑了。不过,要想从脚本小子变成黑客大神,这个方向越往后,需要学习和掌握的东西就会越来越多以下是网络渗透需要学习的内容:  
 ![在这里插入图片描述](https://img-blog.csdnimg.cn/7a04c5d629f1415a9e35662316578e07.png#pic_center)



  • 5
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值