golang mongo事务操作

在 golang 中，我们可以使用 mongo-go-driver 库来进行 MongoDB 的操作。以下是使用 mongo-go-driver 库进行 MongoDB 事务操作的步骤：

1.创建 MongoClient

首先，我们需要创建一个 MongoClient 对象来连接 MongoDB 服务：

client, err := mongo.Connect(context.Background(), options.Client().ApplyURI("mongodb://localhost:27017"))
if err != nil {
    log.Fatal(err)
}
defer func() {
    if err = client.Disconnect(context.Background()); err != nil {
        log.Fatal(err)
    }
}()

2.创建事务

然后，我们需要创建一个 MongoSession 对象来进行事务操作：

session, err := client.StartSession()
if err != nil {
    log.Fatal(err)
}
defer session.EndSession(context.Background())
txnOpts := options.Transaction().SetReadConcern(readconcern.Majority()).SetWriteConcern(writeconcern.New(writeconcern.WMajority()))
txnFunc := func(sessCtx mongo.SessionContext) (interface{}, error) {
    ...
}
result, err := session.WithTransaction(context.Background(), txnFunc, txnOpts)
if err != nil {
    log.Fatal(err)
}

txnFuncWithTransaction()

3.事务操作

在进入事务函数前，需要将需要执行事务操作的 Collection 绑定到 Session 上：

collection := client.Database("db").Collection("coll")
if err = session.PinnedServer().Bind(collection); err != nil {
    log.Fatal(err)
}

在事务函数中，我们可以对数据库进行操作，如插入、更新、删除等操作，例如：

result, err := collection.InsertOne(sessCtx, bson.M{"name": "golang"})
if err != nil {
    return nil, err
}

完整代码如下：

client, err := mongo.Connect(context.Background(), options.Client().ApplyURI("mongodb://localhost:27017"))
if err != nil {
    log.Fatal(err)
}
defer func() {
    if err = client.Disconnect(context.Background()); err != nil {
        log.Fatal(err)
    }
}()
session, err := client.StartSession()
if err != nil {
    log.Fatal(err)
}
defer session.EndSession(context.Background())
txnOpts := options.Transaction().SetReadConcern(readconcern.Majority()).SetWriteConcern(writeconcern.New(writeconcern.WMajority()))
txnFunc := func(sessCtx mongo.SessionContext) (interface{}, error) {
    collection := client.Database("db").Collection("coll")
    if err = sessCtx.PinnedServer().Bind(collection); err != nil {
        log.Fatal(err)
    }
    result, err := collection.InsertOne(sessCtx, bson.M{"name": "golang"})
    if err != nil {
        return nil, err
    }
    return result, nil
}
result, err := session.WithTransaction(context.Background(), txnFunc, txnOpts)
if err != nil {
    log.Fatal(err)
}
log.Println(result)