1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
# frozen_string_literal: true
module Sidekiq
class JobLogger
def initialize(logger = Sidekiq.logger)
@logger = logger
end
def call(item, queue)
start = ::Process.clock_gettime(::Process::CLOCK_MONOTONIC)
@logger.info("start")
yield
with_elapsed_time_context(start) do
@logger.info("done")
end
rescue Exception
with_elapsed_time_context(start) do
@logger.info("fail")
end
raise
end
def prepare(job_hash, &block)
level = job_hash["log_level"]
if level
@logger.log_at(level) do
Sidekiq::Context.with(job_hash_context(job_hash), &block)
end
else
Sidekiq::Context.with(job_hash_context(job_hash), &block)
end
end
def job_hash_context(job_hash)
# If we're using a wrapper class, like ActiveJob, use the "wrapped"
# attribute to expose the underlying thing.
h = {
class: job_hash["wrapped"] || job_hash["class"],
jid: job_hash["jid"],
}
h[:bid] = job_hash["bid"] if job_hash["bid"]
h[:tags] = job_hash["tags"] if job_hash["tags"]
h
end
def with_elapsed_time_context(start, &block)
Sidekiq::Context.with(elapsed_time_context(start), &block)
end
def elapsed_time_context(start)
{elapsed: elapsed(start).to_s}
end
private
def elapsed(start)
(::Process.clock_gettime(::Process::CLOCK_MONOTONIC) - start).round(3)
end
end
end
|
