我正在使用 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)
在这两个示例中,如果出现错误,它们都不会回滚。我知道这是一个演示样本。但是当我在代码中执行相同操作时,效果很好。
发生错误时是否可以省略回滚(驱动程序是否处理)?还是我错过了什么?
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
}