我正在使用golang驱动程序go.mongodb.org/mongo-driver/mongo学习mongodb事务。我正在 github 上关注这个 so 答案和这个示例。

@simagix 给出的示例代码：

if session, err = client.StartSession(); err != nil {
    t.Fatal(err)
}
if err = session.StartTransaction(); err != nil {
    t.Fatal(err)
}
if err = mongo.WithSession(ctx, session, func(sc mongo.SessionContext) error {
    if result, err = collection.UpdateOne(sc, bson.M{"_id": id}, update); err != nil {
        t.Fatal(err)
    }
    if result.MatchedCount != 1 || result.ModifiedCount != 1 {
        t.Fatal("replace failed, expected 1 but got", result.MatchedCount)
    }

    // more interrelated operations ...

    if err = session.CommitTransaction(sc); err != nil {
        t.Fatal(err)
    }
    return nil
}); err != nil {
    t.Fatal(err)
}
session.EndSession(ctx)

在这两个示例中，如果出现错误，它们都不会回滚。我知道这是一个演示样本。但是当我在代码中执行相同操作时，它工作得很好。

在发生错误时是否可以省略回滚（驱动程序是否处理它）？或者我错过了什么？

正确答案

mongo.withsession() 确实不假设任何活动事务，它“仅”有助于在给定会话下运行回调。因此，如果您希望它作为启动事务的一部分执行，您应该自己处理提交和中止。这样可以进行更精细的控制。

但是，如果您打算将回调作为事务运行，则使用 session.withtransaction() 因为它以透明的方式处理事务及其生命周期：它创建事务并根据回调返回的错误提交或中止它。作为额外的功能，它还可以处理重试。正如其文档还指出的那样：

如果回调失败，驱动程序将调用aborttransaction。

这是一个如何在事务中正确执行回调的简单示例：

var docToInsert, idToUpdate, updateDoc any

func doInTransactionExample(ctx context.Context, client *mongo.Client) error {
    sess, err := client.StartSession(options.Session().SetDefaultReadConcern(readconcern.Majority()))
    if err != nil {
        return fmt.Errorf("client.StartSession() error: %w", err)
    }
    defer sess.EndSession(ctx)

    result, err := sess.WithTransaction(
        ctx,
        func(sessCtx mongo.SessionContext) (any, error) {
            // sessCtx must be used as context.Context for all operations to be run in the transaction.
            var ctx context.Context = sessCtx // Shadow ctx on purpose!

            c := client.Database("foo").Collection("bar")

            // Insert example
            if _, err := c.InsertOne(ctx, docToInsert); err != nil {
                return nil, fmt.Errorf("InsertOne() failed: %w", err)
            }

            // Update example
            if ur, err := c.UpdateByID(ctx, idToUpdate, updateDoc); err != nil {
                return nil, fmt.Errorf("UpdateByID() failed: %w", err)
            } else {
                if ur.MatchedCount == 0 {
                    return nil, fmt.Errorf("UpdateByID() failed: %w", mongo.ErrNoDocuments)
                }
            }

            return "arbitrary-result-to-return", nil
        },
        options.Transaction().SetReadPreference(readpref.PrimaryPreferred()),
    )

    if err != nil {
        return fmt.Errorf("sess.WithTransaction() error: %w", err)
    }

    _ = result // Not using result

    return nil
}