[Host.Outbox] Support for UUIDv7 values and MSSQL side ID generation,…

… flexible PK types, refactor Signed-off-by: Tomasz Maruszak <[email protected]>
zarusz · Oct 24, 2024 · c492eb2 · c492eb2
1 parent 19fed04
commit c492eb2
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/SlimMessageBus.Host.Sql.Common/Transactions/AbstractSqlTransactionService.cs b/src/SlimMessageBus.Host.Sql.Common/Transactions/AbstractSqlTransactionService.cs
@@ -31,14 +31,19 @@ protected async virtual ValueTask DisposeAsyncCore()
 
     public async virtual Task BeginTransaction()
     {
+        if (_transactionCompleted)
+        {
+            throw new MessageBusException("Transaction was completed already");
+        }
+
         if (_transactionCount == 0)
         {
             // Start transaction
             await OnBeginTransaction();
             _transactionFailed = false;
             _transactionCompleted = false;
-            _transactionCount++;
         }
+        _transactionCount++;
     }
 
     private async Task TryCompleteTransaction(bool transactionFailed = false)
@@ -59,6 +64,7 @@ private async Task TryCompleteTransaction(bool transactionFailed = false)
         if (!_transactionCompleted && (_transactionCount == 0 || transactionFailed))
         {
             _transactionCompleted = true;
+            _transactionCount = 0;
             await OnCompleteTransaction(_transactionFailed);
         }
     }