From 8675975ba212be389624ed165c287e9988e916db Mon Sep 17 00:00:00 2001 From: ksss Date: Wed, 30 Nov 2016 11:02:02 +0900 Subject: Implement Enumerable::Lazy#to_enum and enum_for --- mrbgems/mruby-enum-lazy/mrblib/lazy.rb | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'mrbgems/mruby-enum-lazy/mrblib') diff --git a/mrbgems/mruby-enum-lazy/mrblib/lazy.rb b/mrbgems/mruby-enum-lazy/mrblib/lazy.rb index 8ce363c6d..c9c3cc5fd 100644 --- a/mrbgems/mruby-enum-lazy/mrblib/lazy.rb +++ b/mrbgems/mruby-enum-lazy/mrblib/lazy.rb @@ -40,6 +40,15 @@ module Enumerable } end + def to_enum(meth=:each, *args, &block) + lz = Lazy.new(self, &block) + lz.obj = self + lz.meth = meth + lz.args = args + lz + end + alias enum_for to_enum + def map(&block) Lazy.new(self){|yielder, val| yielder << block.call(val) -- cgit v1.2.3