Add special Phlex::CSV handling for ActiveRecord::Relation collections

Author: willcosgrove

lib/phlex/rails.rb

@@ -17,6 +17,10 @@ module Rails
 		end
 	end
 
+	class CSV
+		prepend Phlex::Rails::CSV::Overrides
+	end
+
 	class SGML
 		extend Phlex::Rails::SGML::ClassMethods
 

lib/phlex/rails/csv/overrides.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Phlex
+	module Rails
+		module CSV
+			module Overrides
+				def each_item(&block)
+					case collection
+					when ActiveRecord::Relation
+						collection.find_each(&block)
+					else
+						super
+					end
+				end
+			end
+		end
+	end
+end