simon/iter
1
0
Fork 0
mirror of https://github.com/c-cube/iter.git synced 2025-12-06 03:05:29 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

bugfix in Sequence.take

Browse source
This commit is contained in:
Simon Cruanes 2013-10-03 21:54:51 +02:00
parent ad23803a4a
commit 049fe898ce
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
sequence.ml
View file

@ -329,11 +329,13 @@ exception ExitSequence
(** Take at most [n] elements from the sequence *)
let take n seq =
let count = ref 0 in
fun k ->
try
seq
(fun x -> if !count < n then begin incr count; k x end
else raise ExitSequence)
if n = 0 then empty
else fun k ->
try
seq (fun x ->
incr count;
k x;
if !count = n then raise ExitSequence)
with ExitSequence -> ()
(** Drop the [n] first elements of the sequence *)