mongodb 事务 golang

在 Golang 中使用 MongoDB 的事务，需要使用 MongoDB Go 驱动程序提供的事务 API。下面是一个简单的示例：

session, err := client.StartSession()
if err != nil {
    log.Fatal(err)
}
defer session.EndSession(context.Background())

callback := func(sessCtx mongo.SessionContext) (interface{}, error) {
    coll1 := client.Database("mydb").Collection("collection1")
    coll2 := client.Database("mydb").Collection("collection2")

    err := sessCtx.StartTransaction()
    if err != nil {
        return nil, err
    }

    // Perform operations on collections inside the transaction
    _, err = coll1.InsertOne(sessCtx, bson.M{"name": "John"})
    if err != nil {
        sessCtx.AbortTransaction(sessCtx)
        return nil, err
    }

    _, err = coll2.InsertOne(sessCtx, bson.M{"name": "Doe"})
    if err != nil {
        sessCtx.AbortTransaction(sessCtx)
        return nil, err
    }

    err = sessCtx.CommitTransaction(sessCtx)
    if err != nil {
        return nil, err
    }

    return "success", nil
}

result, err := session.WithTransaction(context.Background(), callback)
if err != nil {
    log.Fatal(err)
}

fmt.Println(result)

在此示例中，我们使用 `StartSession()` 方法创建一个 MongoDB 会话对象。然后，我们定义一个回调函数 `callback`，该函数包含了我们想要在事务中执行的操作。在回调函数内部，我们首先使用 `StartTransaction()` 方法开始一个事务。然后，我们执行一些操作（例如，在两个集合中插入文档），并在操作完成后调用 `CommitTransaction()` 方法提交事务。如果事务执行过程中出现任何错误，我们可以调用 `AbortTransaction()` 方法回滚事务。

最后，我们使用 `WithTransaction()` 方法来执行回调函数。如果事务执行成功，`WithTransaction()` 方法将返回回调函数的结果，否则将返回一个错误。