v/vlib/log/log.v

141 lines
2.6 KiB
V
Raw Normal View History

2019-06-26 02:14:38 +02:00
module log
import (
os
time
term
filepath
)
2019-06-26 02:14:38 +02:00
2019-11-25 05:50:59 +01:00
pub enum LogLevel {
2019-12-01 10:50:13 +01:00
fatal = 1
2019-11-25 05:50:59 +01:00
error
2019-12-01 10:50:13 +01:00
warn
2019-11-25 05:50:59 +01:00
info
debug
}
fn tag(l LogLevel) string {
return match l {
.fatal { term.red('FATAL') }
.error { term.red('ERROR') }
.warn { term.yellow('WARN ') }
.info { term.white('INFO ') }
.debug { term.blue('DEBUG') }
else { ' ' }
2019-11-25 05:50:59 +01:00
}
}
2019-10-28 16:53:02 +01:00
pub const (
2019-11-25 05:50:59 +01:00
FATAL = 1
ERROR = 2
WARN = 3
INFO = 4
DEBUG = 5
2019-06-26 02:14:38 +02:00
)
interface Logger {
2019-11-25 05:50:59 +01:00
fatal(s string)
error(s string)
warn(s string)
info(s string)
debug(s string)
}
2019-10-28 16:53:02 +01:00
pub struct Log {
2019-12-01 10:50:13 +01:00
mut:
level LogLevel
output_label string
ofile os.File
output_to_file bool
pub mut:
2019-12-01 10:50:13 +01:00
output_file_name string
2019-06-26 02:14:38 +02:00
}
pub fn (l mut Log) set_level(level int) {
2019-11-25 05:50:59 +01:00
l.level = match level {
FATAL { LogLevel.fatal }
ERROR { LogLevel.error }
2019-12-01 10:50:13 +01:00
WARN { LogLevel.warn }
2019-11-25 05:50:59 +01:00
INFO { LogLevel.info }
DEBUG { LogLevel.debug }
else { .debug }
}
}
pub fn (l mut Log) set_output_level(level LogLevel) {
2019-11-25 05:50:59 +01:00
l.level = level
2019-06-26 02:14:38 +02:00
}
2019-12-01 10:50:13 +01:00
pub fn (l mut Log) set_full_logpath(full_log_path string) {
rlog_file := os.realpath( full_log_path )
2019-12-23 11:09:22 +01:00
l.set_output_label( filepath.filename( rlog_file ) )
l.set_output_path( filepath.basedir( rlog_file ) )
}
2019-12-01 10:50:13 +01:00
pub fn (l mut Log) set_output_label(label string){
2019-11-25 05:50:59 +01:00
l.output_label = label
2019-07-24 17:50:29 +02:00
}
2019-12-01 10:50:13 +01:00
pub fn (l mut Log) set_output_path(output_file_path string) {
if l.ofile.is_opened() { l.ofile.close() }
l.output_to_file = true
l.output_file_name = filepath.join( os.realpath( output_file_path ) , l.output_label )
ofile := os.open_append( l.output_file_name ) or {
2019-12-01 10:50:13 +01:00
panic('error while opening log file ${l.output_file_name} for appending')
}
l.ofile = ofile
2019-11-25 05:50:59 +01:00
}
pub fn (l mut Log) close() {
2019-12-01 10:50:13 +01:00
l.ofile.close()
}
fn (l mut Log) log_file(s string, level LogLevel) {
2019-11-25 05:50:59 +01:00
timestamp := time.now().format_ss()
e := tag(level)
2019-12-01 10:50:13 +01:00
l.ofile.writeln('$timestamp [$e] $s')
2019-11-25 05:50:59 +01:00
}
fn (l &Log) log_cli(s string, level LogLevel) {
2019-11-25 05:50:59 +01:00
f := tag(level)
t := time.now()
println('[$f ${t.format_ss()}] $s')
2019-07-24 17:50:29 +02:00
}
fn (l mut Log) send_output(s &string, level LogLevel) {
2019-12-01 10:50:13 +01:00
if l.output_to_file {
l.log_file(s, level)
} else {
l.log_cli(s, level)
2019-11-25 05:50:59 +01:00
}
2019-06-26 02:14:38 +02:00
}
2019-12-01 10:50:13 +01:00
pub fn (l mut Log) fatal(s string){
if l.level < .fatal { return }
l.send_output(s, .fatal)
l.ofile.close()
panic('$l.output_label: $s')
}
2019-11-25 05:50:59 +01:00
pub fn (l mut Log) error(s string) {
2019-12-01 10:50:13 +01:00
if l.level < .error { return }
l.send_output(s, .error)
2019-06-26 02:14:38 +02:00
}
pub fn (l mut Log) warn(s string) {
2019-12-01 10:50:13 +01:00
if l.level < .warn { return }
l.send_output(s, .warn)
2019-06-26 02:14:38 +02:00
}
pub fn (l mut Log) info(s string) {
2019-12-01 10:50:13 +01:00
if l.level < .info { return }
l.send_output(s, .info)
2019-06-26 02:14:38 +02:00
}
pub fn (l mut Log) debug(s string) {
if l.level < .debug { return }
2019-12-01 10:50:13 +01:00
l.send_output(s, .debug)
2019-07-16 17:59:07 +02:00
}
2019-12-01 10:50:13 +01:00