debian-mirror-gitlab/lib/gitlab/lograge/custom_options.rb

62 lines
2.4 KiB
Ruby
Raw Normal View History

2020-04-08 14:13:33 +05:30
# frozen_string_literal: true
module Gitlab
module Lograge
module CustomOptions
2020-05-24 23:13:21 +05:30
include ::Gitlab::Logging::CloudflareHelper
2020-04-08 14:13:33 +05:30
LIMITED_ARRAY_SENTINEL = { key: 'truncated', value: '...' }.freeze
IGNORE_PARAMS = Set.new(%w(controller action format)).freeze
def self.call(event)
params = event
.payload[:params]
.each_with_object([]) { |(k, v), array| array << { key: k, value: v } unless IGNORE_PARAMS.include?(k) }
payload = {
time: Time.now.utc.iso8601(3),
params: Gitlab::Utils::LogLimitedArray.log_limited_array(params, sentinel: LIMITED_ARRAY_SENTINEL),
remote_ip: event.payload[:remote_ip],
user_id: event.payload[:user_id],
username: event.payload[:username],
2020-06-23 00:09:42 +05:30
ua: event.payload[:ua]
2020-04-08 14:13:33 +05:30
}
2020-06-23 00:09:42 +05:30
add_db_counters!(payload)
2020-04-08 14:13:33 +05:30
2020-05-24 23:13:21 +05:30
payload.merge!(event.payload[:metadata]) if event.payload[:metadata]
2020-04-08 14:13:33 +05:30
::Gitlab::InstrumentationHelper.add_instrumentation_data(payload)
2020-06-23 00:09:42 +05:30
payload[:queue_duration_s] = event.payload[:queue_duration_s] if event.payload[:queue_duration_s]
2020-04-08 14:13:33 +05:30
payload[:response] = event.payload[:response] if event.payload[:response]
payload[:etag_route] = event.payload[:etag_route] if event.payload[:etag_route]
2020-06-23 00:09:42 +05:30
payload[Labkit::Correlation::CorrelationId::LOG_KEY] = event.payload[Labkit::Correlation::CorrelationId::LOG_KEY] || Labkit::Correlation::CorrelationId.current_id
2020-04-08 14:13:33 +05:30
if cpu_s = Gitlab::Metrics::System.thread_cpu_duration(::Gitlab::RequestContext.instance.start_thread_cpu_time)
2020-04-22 19:07:51 +05:30
payload[:cpu_s] = cpu_s.round(2)
2020-04-08 14:13:33 +05:30
end
2020-05-24 23:13:21 +05:30
CLOUDFLARE_CUSTOM_HEADERS.each do |_, value|
payload[value] = event.payload[value] if event.payload[value]
end
2020-04-08 14:13:33 +05:30
# https://github.com/roidrage/lograge#logging-errors--exceptions
exception = event.payload[:exception_object]
::Gitlab::ExceptionLogFormatter.format!(exception, payload)
payload
end
2020-06-23 00:09:42 +05:30
def self.add_db_counters!(payload)
current_transaction = Gitlab::Metrics::Transaction.current
if current_transaction
payload[:db_count] = current_transaction.get(:db_count, :counter).to_i
payload[:db_write_count] = current_transaction.get(:db_write_count, :counter).to_i
payload[:db_cached_count] = current_transaction.get(:db_cached_count, :counter).to_i
end
end
private_class_method :add_db_counters!
2020-04-08 14:13:33 +05:30
end
end
end