diff --git a/lib/oink/middleware.rb b/lib/oink/middleware.rb index f3d1799..173fa94 100644 --- a/lib/oink/middleware.rb +++ b/lib/oink/middleware.rb @@ -20,7 +20,6 @@ def call(env) log_memory log_activerecord log_completed - reset_objects_instantiated [status, headers, body] end @@ -48,13 +47,14 @@ def log_activerecord sorted_list = Oink::HashUtils.to_sorted_array(ActiveRecord::Base.instantiated_hash) sorted_list.unshift("Total: #{ActiveRecord::Base.total_objects_instantiated}") @logger.info("Instantiation Breakdown: #{sorted_list.join(' | ')}") + reset_objects_instantiated end end private def reset_objects_instantiated - ActiveRecord::Base.reset_instance_type_count if @instruments.include?(:activerecord) + ActiveRecord::Base.reset_instance_type_count end end