Reworked project
This commit is contained in:
+29
-94
@@ -1,120 +1,55 @@
|
|||||||
require "time"
|
require "log"
|
||||||
require "colorize"
|
|
||||||
|
|
||||||
module Logger
|
module Logger
|
||||||
|
macro included
|
||||||
annotation LoggerWatcher
|
annotation LogContext
|
||||||
end
|
end
|
||||||
|
|
||||||
struct LOG_LEVEL
|
Log = ::Log
|
||||||
getter value : Int32
|
|
||||||
getter color : Symbol
|
|
||||||
delegate in?, to: @value
|
|
||||||
|
|
||||||
def initialize(@value : Int32, @color : Symbol)
|
{% verbatim do %}
|
||||||
|
macro finished
|
||||||
|
{% defined_level_methods = [] of MacroId %}
|
||||||
|
{% levels = %i[ debug info notice error fatal ] %}
|
||||||
|
{% for level in levels %}
|
||||||
|
{% defined_level_methods << level if @type.has_method? level %}
|
||||||
|
{% end %}
|
||||||
|
{% raise "ERROR : You can't include ::Logger when you have the following defined as instance methods: #{defined_level_methods.join(", ").id}" unless defined_level_methods.empty? %}
|
||||||
end
|
end
|
||||||
|
|
||||||
def to_s
|
macro log_with_context(level, message)
|
||||||
LOG_LEVELS.keys[@value].to_s.downcase
|
{% if @type.class? %}
|
||||||
end
|
{% if vars = @type.instance_vars.select(&.annotation(LogContext)) %}
|
||||||
|
Log.with_context( {% for var in vars %} {{var.name}}: @{{var.name}}, {% end %} ) do
|
||||||
{% for op, i in %w[ < > == <= >= ] %}
|
Log.{{level}} { [ generate_log_line, {{message}}].join(" : ") }
|
||||||
def {{op.id}} (other : LOG_LEVEL)
|
|
||||||
@value {{op.id}} other.value
|
|
||||||
end
|
end
|
||||||
|
{% else %}
|
||||||
|
Log.{{level}} { [ generate_log_line, {{message}}].join(" : ") }
|
||||||
|
{% end %}
|
||||||
|
{% else %}
|
||||||
|
Log.{{level}} { [ generate_log_line, {{message}}].join(" : ") }
|
||||||
{% end %}
|
{% end %}
|
||||||
end
|
end
|
||||||
|
|
||||||
LOG_LEVELS = {
|
|
||||||
"DEBUG" => LOG_LEVEL.new(value: 0, color: :green),
|
|
||||||
"INFO" => LOG_LEVEL.new(value: 1, color: :blue),
|
|
||||||
"WARN" => LOG_LEVEL.new(value: 2, color: :yellow),
|
|
||||||
"ERROR" => LOG_LEVEL.new(value: 3, color: :red),
|
|
||||||
"FATAL" => LOG_LEVEL.new(value: 4, color: :red)
|
|
||||||
}
|
|
||||||
|
|
||||||
@@min_level : LOG_LEVEL = LOG_LEVELS["INFO"]
|
|
||||||
@@max_level : LOG_LEVEL = LOG_LEVELS["FATAL"]
|
|
||||||
@@log_file : String = ""
|
|
||||||
@@debug : Bool = false
|
|
||||||
|
|
||||||
LOG_MUTEX = ::Mutex.new
|
|
||||||
|
|
||||||
macro generate_log_line
|
macro generate_log_line
|
||||||
{% verbatim do %}
|
{% verbatim do %}
|
||||||
{% if @type && @def %}
|
{% if @type && @def %}
|
||||||
"{{(@type.module? ? "Module" : "Class").id}}({{@type.name.id}})" + " : " + "Method({{@def.name}})" + " : "
|
"{{(@type.module? ? "Module" : "Class").id}}({{@type.name}})" + " : " + "Method({{@def.name}})"
|
||||||
{% elsif @type %}
|
{% elsif @type %}
|
||||||
"{{(@type.module? ? "Module" : "Class").id}}({{@type.name.id}})" + " : "
|
"{{(@type.module? ? "Module" : "Class").id}}({{@type.name}})"
|
||||||
{% elsif @def %}
|
{% elsif @def %}
|
||||||
"Method({{@def.name}})" + " : "
|
"Method({{@def.name}})"
|
||||||
{% else %}
|
{% else %}
|
||||||
{{message}}
|
{{message}}
|
||||||
{% end %}
|
{% end %}
|
||||||
{% end %}
|
{% end %}
|
||||||
end
|
end
|
||||||
|
|
||||||
{% for level in LOG_LEVELS.keys %}
|
{% for level in %i[ trace debug info notice error fatal ].map(&.id) %}
|
||||||
macro {{level.downcase.id}}(message, *args)
|
macro {{level.downcase}}(message, *args)
|
||||||
::Logger.log({{level}}, Logger.generate_log_line + \{{message}}, \{{*args}})
|
log_with_context {{level}}, (\{{message}} \{% unless args.empty? %} % \{{args}} \{% end %})
|
||||||
end
|
end
|
||||||
{% end %}
|
{% end %}
|
||||||
|
{% end %}
|
||||||
protected def self.synchronize(&block : -> )
|
|
||||||
LOG_MUTEX.synchronize do
|
|
||||||
yield
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.debug? : Bool
|
|
||||||
LOG_MUTEX.synchronize do
|
|
||||||
@@debug
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.min=(selection : String) : Nil
|
|
||||||
Logger.debug "Entering with #{selection}"
|
|
||||||
raise "ERROR : {{level.id}}= : Invalid level" unless _level = LOG_LEVELS[selection.upcase]
|
|
||||||
synchronize {
|
|
||||||
@@min_level = _level
|
|
||||||
}
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.min : String
|
|
||||||
@@min_level.to_s.downcase
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.log_file=(@@log_file : String)
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.valid?(level : LOG_LEVEL) : Bool
|
|
||||||
level.in?(@@min_level.value..@@max_level.value)
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.debug= (state : Bool) : Nil
|
|
||||||
synchronize {
|
|
||||||
@@debug = state
|
|
||||||
}
|
|
||||||
Logger.debug "Debug set to #{state}"
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.log(level : String, message : String, *args) : Nil
|
|
||||||
_level = LOG_LEVELS[level]
|
|
||||||
message = message % args unless args.empty?
|
|
||||||
if valid?(_level) || debug?
|
|
||||||
synchronize {
|
|
||||||
time = Time.local
|
|
||||||
message = [time, level.to_s.colorize(_level.color), message].join(" : ")
|
|
||||||
|
|
||||||
STDERR.puts message
|
|
||||||
|
|
||||||
unless @@log_file.empty?
|
|
||||||
File.open(@@log_file, "a") do |file|
|
|
||||||
file.puts message
|
|
||||||
end
|
|
||||||
end
|
|
||||||
raise "ERROR : Fatal error detected (#{message})" if level == "FATAL"
|
|
||||||
}
|
|
||||||
end
|
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|||||||
Reference in New Issue
Block a user