Fix and simplify code for TaskSeq.drop

fsprojects · Dec 19, 2023 · dd5b085 · dd5b085
1 parent 6e6821e
commit dd5b085
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/FSharp.Control.TaskSeq/TaskSeqInternal.fs b/src/FSharp.Control.TaskSeq/TaskSeqInternal.fs
@@ -672,14 +672,16 @@ module internal TaskSeqInternal =
                     // skip, or stop looping if we reached the end
                     while cont do
                         pos <- pos + 1
-                        let! moveNext = e.MoveNextAsync()
-                        cont <- moveNext && pos <= count
+
+                        if pos < count then
+                            let! moveNext = e.MoveNextAsync()
+                            cont <- moveNext
+                        else
+                            cont <- false
 
                     // return the rest
-                    while cont do
+                    while! e.MoveNextAsync() do
                         yield e.Current
-                        let! moveNext = e.MoveNextAsync()
-                        cont <- moveNext
 
                 }
         | Take ->