在发生错误时省略 mongodb 事务回滚是一个好习惯吗

问题描述 投票:0回答:1

我正在使用 golang 驱动程序学习 mongodb 事务

go.mongodb.org/mongo-driver/mongo
。我正在关注 this SO 答案github 上的这个示例

@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)

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

发生错误时是否可以省略回滚(驱动程序是否处理)?还是我错过了什么?

mongodb go transactions mongo-go
1个回答
1
投票

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
}
© www.soinside.com 2019 - 2024. All rights reserved.