From 1d4bd229b898671328c2a942b04f08065c640c28 Mon Sep 17 00:00:00 2001 From: Akinori MUSHA Date: Wed, 5 Jun 2019 20:39:21 +0900 Subject: Implement Enumerator::Lazy#eager [Feature #15901] --- enumerator.c | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'enumerator.c') diff --git a/enumerator.c b/enumerator.c index 62534ee241..7a0a17b52f 100644 --- a/enumerator.c +++ b/enumerator.c @@ -1893,6 +1893,26 @@ lazy_to_enum(int argc, VALUE *argv, VALUE self) return lazy; } +static VALUE +lazy_eager_size(VALUE self, VALUE args, VALUE eobj) +{ + return enum_size(self); +} + +/* + * call-seq: + * lzy.eager -> enum + * + * Returns a non-lazy Enumerator converted from the lazy enumerator. + */ + +static VALUE +lazy_eager(VALUE self) +{ + return enumerator_init(enumerator_allocate(rb_cEnumerator), + self, sym_each, 0, 0, lazy_eager_size, Qnil); +} + static VALUE lazyenum_yield(VALUE proc_entry, struct MEMO *result) { @@ -3741,6 +3761,7 @@ InitVM_Enumerator(void) rb_define_method(rb_cLazy, "initialize", lazy_initialize, -1); rb_define_method(rb_cLazy, "to_enum", lazy_to_enum, -1); rb_define_method(rb_cLazy, "enum_for", lazy_to_enum, -1); + rb_define_method(rb_cLazy, "eager", lazy_eager, 0); rb_define_method(rb_cLazy, "map", lazy_map, 0); rb_define_method(rb_cLazy, "collect", lazy_map, 0); rb_define_method(rb_cLazy, "flat_map", lazy_flat_map, 0); -- cgit v1.2.3