Merge pull request #53 from Evodim/fixes/entityTableClient-take-not-b…

…reaked Take extension not breaked
Evodim · May 22, 2023 · 2b76486 · 2b76486
2 parents d3b1f8a + 8e59ab8
commit 2b76486
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/Azure.EntityServices.Tables/Extensions/IAsyncEnumerableExtensions.cs b/src/Azure.EntityServices.Tables/Extensions/IAsyncEnumerableExtensions.cs
@@ -128,14 +128,11 @@ public static async IAsyncEnumerable<IEnumerable<T>> TakeAsync<T>(this IAsyncEnu
                 {
                     yield return asyncEntity;
                 }
-                else 
-                if (took >= take)
+                else
                 {
-                    yield return asyncEntity.Take(asyncEntity.Count()- (took - take) );
-
-                }
-                else 
-                yield break;
+                    yield return asyncEntity.Take(asyncEntity.Count() - (took - take));
+                    yield break;
+                } 
             }
         }
     }