From a9c056f1ed254c99152134235bd973ec503f64f2 Mon Sep 17 00:00:00 2001
From: jeg2 <jeg2@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Thu, 26 May 2011 15:35:06 +0000
Subject: * lib/csv.rb: Enhance each() to support Enumerator.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@31740 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 lib/csv.rb | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'lib')

diff --git a/lib/csv.rb b/lib/csv.rb
index 48c2e97eda..c7649b8842 100644
--- a/lib/csv.rb
+++ b/lib/csv.rb
@@ -1787,8 +1787,12 @@ class CSV
   # The data source must be open for reading.
   #
   def each
-    while row = shift
-      yield row
+    if block_given?
+      while row = shift
+        yield row
+      end
+    else
+      to_enum
     end
   end
 
-- 
cgit v1.2.3