module Sidekiq::Logging

def self.logger
  @logger ||= begin
    log = Logger.new(STDOUT)
    log.level = Logger::INFO
    log.formatter = Pretty.new
    log
  end
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