From 2b2183d31dc44af9113ed1d7f82dcdc20e61483b Mon Sep 17 00:00:00 2001
From: Ahmed Yasin Koculu <koculu@gmail.com>
Date: Sun, 1 Sep 2024 02:20:33 +0200
Subject: [PATCH] Update sample.

---
 README.md                                  | 54 +++++++++-------------
 src/ZoneTree/docs/ZoneTree/README-NUGET.md | 54 +++++++++-------------
 2 files changed, 42 insertions(+), 66 deletions(-)

diff --git a/README.md b/README.md
index 7c735ed..6f1719b 100644
--- a/README.md
+++ b/README.md
@@ -402,54 +402,42 @@ public async Task<bool> ExecuteTransactionWithRetryAsync(ZoneTreeFactory<int, in
 {
     using var zoneTree = zoneTreeFactory.OpenOrCreateTransactional();
     var transactionId = zoneTree.BeginTransaction();
-    int retryCount = 0;
 
-    while (retryCount <= maxRetries)
-    {
-        var result = zoneTree.UpsertNoThrow(transactionId, 1, 100);
-        if (result.IsAborted)
-        {
-            // Abort the transaction and return false on failure.
-            return false;
-        }
+    // Execute the operations within the transaction
+    var result = zoneTree.UpsertNoThrow(transactionId, 1, 100);
+    if (result.IsAborted)
+        return false; // Abort the transaction and return false on failure.
 
-        result = zoneTree.UpsertNoThrow(transactionId, 2, 200);
-        if (result.IsAborted)
-        {
-            // Abort the transaction and return false on failure.
-            return false;
-        }
+    result = zoneTree.UpsertNoThrow(transactionId, 2, 200);
+    if (result.IsAborted)
+        return false; // Abort the transaction and return false on failure.
 
+    // Retry only the prepare step
+    int retryCount = 0;
+    while (retryCount <= maxRetries)
+    {
         var prepareResult = zoneTree.PrepareNoThrow(transactionId);
         if (prepareResult.IsAborted)
-        {
-            // Abort the transaction and return false on failure.
-            return false;
-        }
+            return false; // Abort the transaction and return false on failure.
 
         if (prepareResult.IsPendingTransactions)
         {
-            // Optionally wait or handle pending transactions
             await Task.Delay(100); // Simple delay before retrying
             retryCount++;
-            continue; // Retry the transaction
+            continue; // Retry the prepare step
         }
 
         if (prepareResult.IsReadyToCommit)
-        {
-            var commitResult = zoneTree.CommitNoThrow(transactionId);
-            if (commitResult.IsAborted)
-            {
-                // Abort the transaction and return false on failure.
-                return false;
-            }
-            // Transaction committed successfully
-            return true;
-        }
+            break; // Exit loop if ready to commit
     }
 
-    // Return false if retries are exhausted without a successful commit.
-    return false;
+    // After successfully preparing, commit the transaction
+    var commitResult = zoneTree.CommitNoThrow(transactionId);
+    if (commitResult.IsAborted)
+        return false; // Abort the transaction and return false on failure.
+
+    // Transaction committed successfully
+    return true;
 }
 ```
 
diff --git a/src/ZoneTree/docs/ZoneTree/README-NUGET.md b/src/ZoneTree/docs/ZoneTree/README-NUGET.md
index c0321d9..4fdb51e 100644
--- a/src/ZoneTree/docs/ZoneTree/README-NUGET.md
+++ b/src/ZoneTree/docs/ZoneTree/README-NUGET.md
@@ -401,54 +401,42 @@ public async Task<bool> ExecuteTransactionWithRetryAsync(ZoneTreeFactory<int, in
 {
     using var zoneTree = zoneTreeFactory.OpenOrCreateTransactional();
     var transactionId = zoneTree.BeginTransaction();
-    int retryCount = 0;
 
-    while (retryCount <= maxRetries)
-    {
-        var result = zoneTree.UpsertNoThrow(transactionId, 1, 100);
-        if (result.IsAborted)
-        {
-            // Abort the transaction and return false on failure.
-            return false;
-        }
+    // Execute the operations within the transaction
+    var result = zoneTree.UpsertNoThrow(transactionId, 1, 100);
+    if (result.IsAborted)
+        return false; // Abort the transaction and return false on failure.
 
-        result = zoneTree.UpsertNoThrow(transactionId, 2, 200);
-        if (result.IsAborted)
-        {
-            // Abort the transaction and return false on failure.
-            return false;
-        }
+    result = zoneTree.UpsertNoThrow(transactionId, 2, 200);
+    if (result.IsAborted)
+        return false; // Abort the transaction and return false on failure.
 
+    // Retry only the prepare step
+    int retryCount = 0;
+    while (retryCount <= maxRetries)
+    {
         var prepareResult = zoneTree.PrepareNoThrow(transactionId);
         if (prepareResult.IsAborted)
-        {
-            // Abort the transaction and return false on failure.
-            return false;
-        }
+            return false; // Abort the transaction and return false on failure.
 
         if (prepareResult.IsPendingTransactions)
         {
-            // Optionally wait or handle pending transactions
             await Task.Delay(100); // Simple delay before retrying
             retryCount++;
-            continue; // Retry the transaction
+            continue; // Retry the prepare step
         }
 
         if (prepareResult.IsReadyToCommit)
-        {
-            var commitResult = zoneTree.CommitNoThrow(transactionId);
-            if (commitResult.IsAborted)
-            {
-                // Abort the transaction and return false on failure.
-                return false;
-            }
-            // Transaction committed successfully
-            return true;
-        }
+            break; // Exit loop if ready to commit
     }
 
-    // Return false if retries are exhausted without a successful commit.
-    return false;
+    // After successfully preparing, commit the transaction
+    var commitResult = zoneTree.CommitNoThrow(transactionId);
+    if (commitResult.IsAborted)
+        return false; // Abort the transaction and return false on failure.
+
+    // Transaction committed successfully
+    return true;
 }
 ```