7a8620a570
The log level can now be set with `log_level` (accepts ints and strings). The log file can now be set with `output` (also accepts `STDOUT`).
60 lines
1.2 KiB
Crystal
60 lines
1.2 KiB
Crystal
require "logger"
|
|
|
|
enum LogLevel
|
|
All = 0
|
|
Trace = 1
|
|
Debug = 2
|
|
Info = 3
|
|
Warn = 4
|
|
Error = 5
|
|
Fatal = 6
|
|
Off = 7
|
|
end
|
|
|
|
class Invidious::LogHandler < Kemal::BaseLogHandler
|
|
def initialize(@io : IO = STDOUT, @level = LogLevel::Debug)
|
|
end
|
|
|
|
def call(context : HTTP::Server::Context)
|
|
elapsed_time = Time.measure { call_next(context) }
|
|
elapsed_text = elapsed_text(elapsed_time)
|
|
|
|
info("#{context.response.status_code} #{context.request.method} #{context.request.resource} #{elapsed_text}")
|
|
|
|
context
|
|
end
|
|
|
|
def puts(message : String)
|
|
@io << message << '\n'
|
|
@io.flush
|
|
end
|
|
|
|
def write(message : String)
|
|
@io << message
|
|
@io.flush
|
|
end
|
|
|
|
def set_log_level(level : String)
|
|
@level = LogLevel.parse(level)
|
|
end
|
|
|
|
def set_log_level(level : LogLevel)
|
|
@level = level
|
|
end
|
|
|
|
{% for level in %w(trace debug info warn error fatal) %}
|
|
def {{level.id}}(message : String)
|
|
if LogLevel::{{level.id.capitalize}} >= @level
|
|
puts("#{Time.utc} [{{level.id}}] #{message}")
|
|
end
|
|
end
|
|
{% end %}
|
|
|
|
private def elapsed_text(elapsed)
|
|
millis = elapsed.total_milliseconds
|
|
return "#{millis.round(2)}ms" if millis >= 1
|
|
|
|
"#{(millis * 1000).round(2)}µs"
|
|
end
|
|
end
|