group db tx

internal/tx/gorm.go

@@ -0,0 +1,19 @@
+package tx
+
+import (
+	"gorm.io/gorm"
+)
+
+func NewGorm(db *gorm.DB) Tx {
+	return &_Gorm{tx: db}
+}
+
+type _Gorm struct {
+	tx *gorm.DB
+}
+
+func (g *_Gorm) Transaction(fn func(tx any) error) error {
+	return g.tx.Transaction(func(tx *gorm.DB) error {
+		return fn(tx)
+	})
+}

internal/tx/mongo.go

@@ -0,0 +1,31 @@
+package tx
+
+import (
+	"Open_IM/pkg/utils"
+	"context"
+	"go.mongodb.org/mongo-driver/mongo"
+)
+
+func NewMongo(client *mongo.Client) CtxTx {
+	return &_Mongo{
+		client: client,
+	}
+}
+
+type _Mongo struct {
+	client *mongo.Client
+}
+
+func (m *_Mongo) Transaction(ctx context.Context, fn func(ctx context.Context) error) error {
+	sess, err := m.client.StartSession()
+	if err != nil {
+		return err
+	}
+	sCtx := mongo.NewSessionContext(ctx, sess)
+	defer sess.EndSession(sCtx)
+	if err := fn(sCtx); err != nil {
+		_ = sess.AbortTransaction(sCtx)
+		return err
+	}
+	return utils.Wrap(sess.CommitTransaction(sCtx), "")
+}

internal/tx/tx.go

@@ -0,0 +1,11 @@
+package tx
+
+import "context"
+
+type Tx interface {
+	Transaction(fn func(tx any) error) error
+}
+
+type CtxTx interface {
+	Transaction(ctx context.Context, fn func(ctx context.Context) error) error
+}