mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
79 lines
1.8 KiB
Ruby
79 lines
1.8 KiB
Ruby
|
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
|