Keep Mongo operations online across Vault credential rotation

backend/shared/database/internal/transaction/factory.go

@@ -1,18 +1,15 @@
 package transactionimp
 
 import (
+	"git.sendico.io/tech/sxchnge/shared/database/source"
 	"git.sendico.io/tech/sxchnge/shared/database/transaction"
-	"go.mongodb.org/mongo-driver/v2/mongo"
 )
 
-type MongoTransactionFactory struct {
-	client *mongo.Client
-}
+type MongoTransactionFactory struct{ database source.Source }
 
-func (mtf *MongoTransactionFactory) CreateTransaction() transaction.Transaction {
-	return Create(mtf.client)
+func (f *MongoTransactionFactory) CreateTransaction() transaction.Transaction {
+	return &MongoTransaction{database: f.database}
 }
-
-func CreateFactory(client *mongo.Client) transaction.Factory {
-	return &MongoTransactionFactory{client: client}
+func CreateSourceFactory(database source.Source) transaction.Factory {
+	return &MongoTransactionFactory{database: database}
 }

backend/shared/database/internal/transaction/transaction.go

@@ -3,32 +3,29 @@ package transactionimp
 import (
 	"context"
 
+	"git.sendico.io/tech/sxchnge/shared/database/source"
 	"git.sendico.io/tech/sxchnge/shared/database/transaction"
-	"go.mongodb.org/mongo-driver/v2/mongo"
+	merrors "git.sendico.io/tech/sxchnge/shared/errors"
 )
 
-type MongoTransaction struct {
-	client *mongo.Client
-}
+type MongoTransaction struct{ database source.Source }
 
-func (mt *MongoTransaction) Execute(ctx context.Context, cb transaction.Callback) (any, error) {
-	// Reuse an outer customer workflow transaction when a token store joins it.
+func (t *MongoTransaction) Execute(ctx context.Context, cb transaction.Callback) (any, error) {
+	if cb == nil {
+		return nil, merrors.InvalidArgument("transaction callback is required")
+	}
+	database, ctx, release, err := t.database.Acquire(ctx)
+	if err != nil {
+		return nil, err
+	}
+	defer release()
 	if transaction.ActiveMongoTransaction(ctx) {
 		return cb(ctx)
 	}
-	session, err := mt.client.StartSession()
+	session, err := database.Client().StartSession()
 	if err != nil {
 		return nil, err
 	}
 	defer session.EndSession(ctx)
-
-	callback := func(c context.Context) (any, error) {
-		return cb(c)
-	}
-
-	return session.WithTransaction(ctx, callback)
-}
-
-func Create(client *mongo.Client) *MongoTransaction {
-	return &MongoTransaction{client: client}
+	return session.WithTransaction(ctx, cb)
 }