debian-mirror-gitlab/lib/gitlab/marginalia/comment.rb

43 lines
1.1 KiB
Ruby
Raw Normal View History

2020-01-01 13:55:28 +05:30
# frozen_string_literal: true
# Module to support correlation_id and additional job details.
module Gitlab
module Marginalia
module Comment
private
def jid
bg_job["jid"] if bg_job.present?
end
def job_class
bg_job["class"] if bg_job.present?
end
def correlation_id
if bg_job.present?
bg_job["correlation_id"]
else
Labkit::Correlation::CorrelationId.current_id
end
end
def bg_job
job = ::Marginalia::Comment.marginalia_job
# We are using 'Marginalia::SidekiqInstrumentation' which does not support 'ActiveJob::Base'.
2020-07-28 23:09:34 +05:30
# Gitlab also uses 'ActionMailer::MailDeliveryJob' which inherits from ActiveJob::Base.
2020-01-01 13:55:28 +05:30
# So below condition is used to return metadata for such jobs.
2020-07-28 23:09:34 +05:30
if job.is_a?(ActionMailer::MailDeliveryJob) || job.is_a?(ActionMailer::DeliveryJob)
2020-01-01 13:55:28 +05:30
{
"class" => job.arguments.first,
"jid" => job.job_id
}
else
job
end
end
end
end
end