module ActiveSupport::TaggedLogging

def self.new(logger)
  # Ensure we set a default formatter so we aren't extending nil!
  logger.formatter ||= ActiveSupport::Logger::SimpleFormatter.new
  logger.formatter.extend Formatter
  logger.extend(self)
end

def flush
  clear_tags!
  super if defined?(super)
end

def tagged(*tags)
  formatter.tagged(*tags) { yield self }
end