b39f01dcdf
Everything that gets logged now has a log level associated with it. The log level can be set with the new `-l` or `--log-level` arguments. The defaul log level is `debug` for now. There aren't many things that get logged but if the logs get spammed in the future it can be set down to `info`.
60 lines
1.2 KiB
Crystal
60 lines
1.2 KiB
Crystal
require "logger"
|
|
|
|
enum LogLevel
|
|
All
|
|
Trace
|
|
Debug
|
|
Info
|
|
Warn
|
|
Error
|
|
Fatal
|
|
Off
|
|
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
|