处理 mongodb 连接时的上下文

golang学习网今天将给大家带来《处理 mongodb 连接时的上下文》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

我通过传递 client 作为参数来使多个 goroutine 共享一个连接。

uri := "mongodb://localhost:27017"
ctx := context.Background()
client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))

go Foo(client)
go Bar(client)

func Foo(client *mongo.Client) {
        // ... 
}

func Bar(client *mongoClient) {
        // ...
}

我对如何处理 ctx 感到困惑。我应该在每次查询数据库时创建一个新的上下文，还是应该像客户端一样重用上下文？

解决方案

这取决于您的 foo 和 bar 方法的行为方式。让我们假设 foo 方法是一个简单的短期 goroutine，它对数据库进行一次查询，您唯一想要的就是检查其父上下文是否未完成或已取消。然后您可以为您的 foo 方法提供父上下文。

func main() {
    uri := "mongodb://localhost:27017"
    ctx := context.background()
    client, err := connect(ctx, uri)

    ctx, cancel := context.withcancel(ctx)

    
    if err != nil {
        panic(err)
    }

    go foo(ctx, client)
    go bar(context.withvalue(ctx, "uri", uri), client)

    // cancel parent context
    cancel()

    time.sleep(5*time.second)
}

func foo(ctx context.context, client *client) {
    fmt.printf("foo: %s\n", ctx.value("uri"))
    select {
        case <- ctx.done():
            err := ctx.err()
            if err != nil {
                // you could switch for the actual reason
                fmt.println("in our case context canceled: ", err)
                return
            }
            fmt.printf("do something...")
    }
}

另一方面，如果 bar 执行一些重要的逻辑并对数据库进行多次调用，您可能需要一个单独的上下文来单独从父上下文中取消它。然后你可以从你的父母那里获得一个新的上下文。

func bar(ctx context.context, client *client) {
    // bar has a non trivial logic and needs a separate cancellation and handling
    ctx, cancelfunc := context.withcancel(ctx)
    fmt.printf("bar: %s\n", ctx.value("uri"))
    
    // cancel derived context
    cancelfunc()

}

我也这样做过

type DB struct {
    client *mongo.Client
}

func (db *DB) GetVideoStream {}

func main() {
    ctx, _ := context.WithTimeout(context.Background(), 60*time.Second)
    client, err := mongo.Connect(ctx, clientOpts)
    db := &DB{client: client}
    go db.GetVideoStream()
    http.HandleFunc("/api/", db.GetVideoStream)
}

您可以使用指针接收器来执行相同的操作。

我对这门语言还是新手

以上就是《处理 mongodb 连接时的上下文》的详细内容，更多关于的资料请关注golang学习网公众号！