module Sidekiq::Logging

def self.initialize_logger(log_target = STDOUT)
  @logger = Logger.new(log_target)
  @logger.level = Logger::INFO
  @logger.formatter = Pretty.new
  @logger
end

def self.logger
  @logger || initialize_logger
end

def self.logger=(log)
  @logger = (log ? log : Logger.new('/dev/null'))
end

def self.with_context(msg)
  begin
    Thread.current[:sidekiq_context] = msg
    yield
  ensure
    Thread.current[:sidekiq_context] = nil
  end
end

def logger
  Sidekiq::Logging.logger
end