debian-mirror-gitlab/lib/gitlab/sidekiq_logging/json_formatter.rb

65 lines
1.5 KiB
Ruby
Raw Normal View History

2019-02-15 15:39:39 +05:30
# frozen_string_literal: true
2020-04-08 14:13:33 +05:30
# This is needed for sidekiq-cluster
require 'json'
2018-05-09 12:01:36 +05:30
module Gitlab
module SidekiqLogging
class JSONFormatter
2020-03-13 15:44:24 +05:30
TIMESTAMP_FIELDS = %w[created_at enqueued_at started_at retried_at failed_at completed_at].freeze
2018-05-09 12:01:36 +05:30
def call(severity, timestamp, progname, data)
output = {
severity: severity,
time: timestamp.utc.iso8601(3)
}
case data
when String
output[:message] = data
when Hash
2020-03-13 15:44:24 +05:30
convert_to_iso8601!(data)
2020-05-24 23:13:21 +05:30
convert_retry_to_integer!(data)
2020-04-22 19:07:51 +05:30
stringify_args!(data)
2018-05-09 12:01:36 +05:30
output.merge!(data)
end
output.to_json + "\n"
end
2020-03-13 15:44:24 +05:30
private
def convert_to_iso8601!(payload)
TIMESTAMP_FIELDS.each do |key|
value = payload[key]
payload[key] = format_time(value) if value.present?
end
end
def format_time(timestamp)
return timestamp unless timestamp.is_a?(Numeric)
Time.at(timestamp).utc.iso8601(3)
end
2020-04-22 19:07:51 +05:30
2020-05-24 23:13:21 +05:30
def convert_retry_to_integer!(payload)
payload['retry'] =
case payload['retry']
when Integer
payload['retry']
when false, nil
0
when true
Sidekiq::JobRetry::DEFAULT_MAX_RETRY_ATTEMPTS
else
-1
end
end
2020-04-22 19:07:51 +05:30
def stringify_args!(payload)
payload['args'] = Gitlab::Utils::LogLimitedArray.log_limited_array(payload['args'].map(&:to_s)) if payload['args']
end
2018-05-09 12:01:36 +05:30
end
end
end