fix: RedshiftConnector.foreach did not work with new exporter

bricolages · Aug 22, 2017 · ad94e0f · ad94e0f
1 parent 368e973
commit ad94e0f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/redshift_connector/exporter.rb b/lib/redshift_connector/exporter.rb
@@ -28,11 +28,11 @@ def Exporter.for_query(**params)
 
     def Exporter.foreach(**params, &block)
       exporter = for_query(**params)
+      bundle = exporter.execute
       begin
-        exporter.execute
-        exporter.bundle.each_row(&block)
+        bundle.each_row(&block)
       ensure
-        exporter.bundle.clear
+        bundle.clear if bundle.respond_to?(:clear)
       end
     end
   end