Better fix for deprecation of Enumerator.new.

rubyworks · Nov 21, 2013 · 7816a87 · 7816a87
1 parent e2cd323
commit 7816a87
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/lib/ostruct2.rb b/lib/ostruct2.rb
@@ -321,12 +321,13 @@ def to_h
   #
   # @return [Enumerator]
   #
-  def to_enum
+  def to_enum(methname=:each!)
     # Why has Ruby 2 deprecated this form?
-    if ::RUBY_VERSION >= "2.0"
-      ::Kernel.instance_method(:to_enum).bind(self).call(:each!)
-    else
-      ::Enumerator.new(self, :each!) 
+    #::Enumerator.new(self, methname) 
+    ::Enumerator.new do |y|
+       __send__(methname) do |*a|
+         y.yield *a
+       end
     end
   end