mirror of
https://github.com/mperham/sidekiq.git
synced 2022-11-09 13:52:34 -05:00
29dca70e24
* Initial work on Sidekiq::Config * Initial work on Sidekiq::Config * reduce dependencies in deploy marks * bare sidekiq and webapp * Modify runtime to work with Capsules * Cleanup * Rename test files to remove test_ prefix * Update test suite and standard rules to be more compliant * Move constant definition outside code, per standard formatting * Loads of changes for introduction of Capsules * Remove Redis adapter abstraction * update capsule overview * Ensure Sidekiq.redis uses the correct pool for jobs running within a Capsule * Use default_capsule for safety * Slow down the beat to halve its Redis overhead * move config fixtures into cfg/ * Add capsule middleware test * use accessor
131 lines
2.9 KiB
Ruby
131 lines
2.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "logger"
|
|
require "time"
|
|
|
|
module Sidekiq
|
|
module Context
|
|
def self.with(hash)
|
|
orig_context = current.dup
|
|
current.merge!(hash)
|
|
yield
|
|
ensure
|
|
Thread.current[:sidekiq_context] = orig_context
|
|
end
|
|
|
|
def self.current
|
|
Thread.current[:sidekiq_context] ||= {}
|
|
end
|
|
|
|
def self.add(k, v)
|
|
current[k] = v
|
|
end
|
|
end
|
|
|
|
module LoggingUtils
|
|
LEVELS = {
|
|
"debug" => 0,
|
|
"info" => 1,
|
|
"warn" => 2,
|
|
"error" => 3,
|
|
"fatal" => 4
|
|
}
|
|
LEVELS.default_proc = proc do |_, level|
|
|
puts("Invalid log level: #{level.inspect}")
|
|
nil
|
|
end
|
|
|
|
LEVELS.each do |level, numeric_level|
|
|
define_method("#{level}?") do
|
|
local_level.nil? ? super() : local_level <= numeric_level
|
|
end
|
|
end
|
|
|
|
def local_level
|
|
Thread.current[:sidekiq_log_level]
|
|
end
|
|
|
|
def local_level=(level)
|
|
case level
|
|
when Integer
|
|
Thread.current[:sidekiq_log_level] = level
|
|
when Symbol, String
|
|
Thread.current[:sidekiq_log_level] = LEVELS[level.to_s]
|
|
when nil
|
|
Thread.current[:sidekiq_log_level] = nil
|
|
else
|
|
raise ArgumentError, "Invalid log level: #{level.inspect}"
|
|
end
|
|
end
|
|
|
|
def level
|
|
local_level || super
|
|
end
|
|
|
|
# Change the thread-local level for the duration of the given block.
|
|
def log_at(level)
|
|
old_local_level = local_level
|
|
self.local_level = level
|
|
yield
|
|
ensure
|
|
self.local_level = old_local_level
|
|
end
|
|
end
|
|
|
|
class Logger < ::Logger
|
|
include LoggingUtils
|
|
|
|
module Formatters
|
|
class Base < ::Logger::Formatter
|
|
def tid
|
|
Thread.current["sidekiq_tid"] ||= (Thread.current.object_id ^ ::Process.pid).to_s(36)
|
|
end
|
|
|
|
def ctx
|
|
Sidekiq::Context.current
|
|
end
|
|
|
|
def format_context
|
|
if ctx.any?
|
|
" " + ctx.compact.map { |k, v|
|
|
case v
|
|
when Array
|
|
"#{k}=#{v.join(",")}"
|
|
else
|
|
"#{k}=#{v}"
|
|
end
|
|
}.join(" ")
|
|
end
|
|
end
|
|
end
|
|
|
|
class Pretty < Base
|
|
def call(severity, time, program_name, message)
|
|
"#{time.utc.iso8601(3)} pid=#{::Process.pid} tid=#{tid}#{format_context} #{severity}: #{message}\n"
|
|
end
|
|
end
|
|
|
|
class WithoutTimestamp < Pretty
|
|
def call(severity, time, program_name, message)
|
|
"pid=#{::Process.pid} tid=#{tid}#{format_context} #{severity}: #{message}\n"
|
|
end
|
|
end
|
|
|
|
class JSON < Base
|
|
def call(severity, time, program_name, message)
|
|
hash = {
|
|
ts: time.utc.iso8601(3),
|
|
pid: ::Process.pid,
|
|
tid: tid,
|
|
lvl: severity,
|
|
msg: message
|
|
}
|
|
c = ctx
|
|
hash["ctx"] = c unless c.empty?
|
|
|
|
Sidekiq.dump_json(hash) << "\n"
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|