blob: b28f4a5ad382389255315b38ef94ae68f593b1e6 (
plain) (
tree)
|
|
module ActiveSupport
# Inspired by the buffered logger idea by Ezra
class BufferedLogger
module Severity
DEBUG = 0
INFO = 1
WARN = 2
ERROR = 3
FATAL = 4
UNKNOWN = 5
end
include Severity
# Set to false to disable the silencer
cattr_accessor :silencer
self.silencer = true
# Silences the logger for the duration of the block.
def silence(temporary_level = ERROR)
if silencer
begin
old_logger_level, self.level = level, temporary_level
yield self
ensure
self.level = old_logger_level
end
else
yield self
end
end
attr_accessor :level, :auto_flushing
attr_reader :buffer
def initialize(log, level = DEBUG)
@level = level
@buffer = ""
@auto_flushing = true
if log.respond_to?(:write)
@log = log
elsif File.exist?(log)
@log = open(log, (File::WRONLY | File::APPEND))
@log.sync = true
else
@log = open(log, (File::WRONLY | File::APPEND | File::CREAT))
@log.sync = true
@log.write("# Logfile created on %s" % [Time.now.to_s])
end
end
for severity in Severity.constants
class_eval <<-EOT
def #{severity.downcase}(message)
return if @level > #{severity}
message << "\\n" unless message[-1] == ?\\n
@buffer << message
flush if auto_flushing
message
end
def #{severity.downcase}?
@level == #{severity}
end
EOT
end
def flush
return if @buffer.size == 0
@log.write(@buffer.slice!(0..-1))
end
def close
flush
@log.close if @log.respond_to?(:close)
@log = nil
end
end
end
|