simon/iter
1
0
Fork 0
mirror of https://github.com/c-cube/iter.git synced 2025-12-06 11:15:32 -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

10
sequence.ml
View file

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