lib/sidekiq/logger.rb
# 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