Merge pull request #20 from derencius/postgres

Postgres Support

lib/polo/collector.rb

@@ -14,10 +14,12 @@ def initialize(base_class, id, dependency_tree={})
     # ActiveSupport::Notifications block and collecting every generate SQL query.
     #
     def collect
-      ActiveSupport::Notifications.subscribed(collector, 'sql.active_record') do
-        base_finder = @base_class.includes(@dependency_tree).where(@base_class.primary_key => @id)
-        collect_sql(@base_class, base_finder.to_sql)
-        base_finder.to_a
+      unprepared_statement do
+        ActiveSupport::Notifications.subscribed(collector, 'sql.active_record') do
+          base_finder = @base_class.includes(@dependency_tree).where(@base_class.primary_key => @id)
+          collect_sql(@base_class, base_finder.to_sql)
+          base_finder.to_a
+        end
       end
 
       @selects.compact.uniq
@@ -33,8 +35,9 @@ def collect
     #
     def collector
       lambda do |name, start, finish, id, payload|
+        return unless payload[:name] =~ /^(.*) Load$/
         begin
-          class_name = payload[:name].gsub(' Load', '').constantize
+          class_name = $1.constantize
           sql = payload[:sql]
           collect_sql(class_name, sql)
         rescue ActiveRecord::StatementInvalid, NameError
@@ -49,5 +52,15 @@ def collect_sql(klass, sql)
         sql: sql
       }
     end
+
+    def unprepared_statement
+      if ActiveRecord::Base.connection.respond_to?(:unprepared_statement)
+        ActiveRecord::Base.connection.unprepared_statement do
+          yield
+        end
+      else
+        yield
+      end
+    end
   end
 end