diff --git a/lib/semlogger.rb b/lib/semlogger.rb index 678e227..79405d2 100644 --- a/lib/semlogger.rb +++ b/lib/semlogger.rb @@ -27,9 +27,19 @@ end end class Semlogger < ::Logger - class CustomType + class Base attr_accessor :logger + def add severity, progname = nil, &block + @logger.add severity, self, progname = nil, &block + end + + ::Semlogger::Severity.constants.each do |severity| + module_eval "def #{severity.downcase}( *a, &e) add #{::Semlogger::Severity.const_get severity}, *a, &e end", __FILE__, __LINE__ + end + end + + class CustomType < Base def initialize name, *obj @name, @obj = name.to_s.to_sym, obj end @@ -37,13 +47,6 @@ class Semlogger < ::Logger def to_semlogger [@name] + @obj end - - def add severity, progname = nil, &block - @logger.add severity, self, progname = nil, &block - end - ::Semlogger::Severity.constants.each do |severity| - module_eval "def #{severity.downcase}( *a, &e) add #{::Semlogger::Severity.const_get severity}, *a, &e end", __FILE__, __LINE__ - end end attr_accessor :logdev, :level, :progname