diff --git a/lib/chitragupta.rb b/lib/chitragupta.rb index 456b96b..3cc4c09 100644 --- a/lib/chitragupta.rb +++ b/lib/chitragupta.rb @@ -57,6 +57,9 @@ def configure_app(app) require "chitragupta/request_log_formatter" config.lograge.enabled = true config.lograge.formatter = RequestLogFormatter::FORMAT + config.lograge.custom_options = lambda do |event| + {:query_count => event.payload[:query_count]} + end end # setting the log_tags to empty array to ensure that the message being generated does not contain the unwanted tags diff --git a/lib/chitragupta/util.rb b/lib/chitragupta/util.rb index 09e7370..6ba7233 100644 --- a/lib/chitragupta/util.rb +++ b/lib/chitragupta/util.rb @@ -64,6 +64,7 @@ def populate_rails_server_data(data, message) data[:data][:response][:view_rendering_duration] = message[:view] rescue nil data[:data][:response][:db_query_duration] = message[:db] rescue nil + data[:data][:response][:db_query_count] = message[:query_count] rescue nil end def populate_ruby_server_data(data, message) diff --git a/lib/chitragupta/version.rb b/lib/chitragupta/version.rb index 94ccffb..832138a 100644 --- a/lib/chitragupta/version.rb +++ b/lib/chitragupta/version.rb @@ -1,3 +1,3 @@ module Chitragupta - VERSION = "0.4.1" + VERSION = "0.4.2" end diff --git a/spec/chitragupta/util_spec.rb b/spec/chitragupta/util_spec.rb index 174a93f..877cdd2 100644 --- a/spec/chitragupta/util_spec.rb +++ b/spec/chitragupta/util_spec.rb @@ -93,7 +93,8 @@ class Console status: nil, duration: nil, view_rendering_duration: nil, - db_query_duration: nil + db_query_duration: nil, + db_query_count: nil } }, meta: { @@ -124,7 +125,8 @@ class Console status: 200, duration: 10.01, view_rendering_duration: 12.34, - db_query_duration: 0.09 + db_query_duration: 0.09, + db_query_count: 10 } }, meta: { @@ -137,7 +139,8 @@ class Console status: 200, duration: 10.01, view: 12.34, - db: 0.09 + db: 0.09, + query_count: 10 } Chitragupta::Util.send(:populate_rails_server_data, data, message) expect(data).to eq(expected_output)