You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.2 KiB
51 lines
1.2 KiB
package dorm
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
)
|
|
|
|
type MongoSessionOptions struct {
|
|
Session mongo.SessionContext // 会话
|
|
Db *mongo.Client // 驱动
|
|
startSession mongo.Session // 开始会话
|
|
}
|
|
|
|
// Begin 开始事务，会同时创建开始会话需要在退出时关闭会话
|
|
func (c *MongoClient) Begin() (ms *MongoSessionOptions, err error) {
|
|
|
|
var ctx = context.Background()
|
|
|
|
ms.Db = c.Db
|
|
|
|
// 开始会话
|
|
ms.startSession, err = ms.Db.StartSession()
|
|
if err != nil {
|
|
return nil, errors.New(fmt.Sprintf("开始会话失败：%v", err))
|
|
}
|
|
|
|
// 会话上下文
|
|
ms.Session = mongo.NewSessionContext(ctx, ms.startSession)
|
|
|
|
// 会话开启事务
|
|
err = ms.startSession.StartTransaction()
|
|
return ms, err
|
|
}
|
|
|
|
// Close 关闭会话
|
|
func (cs *MongoSessionOptions) Close(ctx context.Context) {
|
|
cs.startSession.EndSession(ctx)
|
|
}
|
|
|
|
// Rollback 回滚事务
|
|
func (cs *MongoSessionOptions) Rollback(ctx context.Context) error {
|
|
return cs.startSession.AbortTransaction(ctx)
|
|
}
|
|
|
|
// Commit 提交事务
|
|
func (cs *MongoSessionOptions) Commit(ctx context.Context) error {
|
|
return cs.startSession.CommitTransaction(ctx)
|
|
}
|