Add files via upload

Author: oxinabox

src/demuxlogger.jl

@@ -0,0 +1,39 @@
+struct DemuxLogger <: AbstractLogger
+	loggers::Vector{AbstractLogger}
+end
+
+"""
+	demux_global_logger!(loggers...; include_current=true)
+
+Sets the global_logger to demux,
+so that messages are sent to all the loggers.
+If `include_current` is true, then messages are also sent to the old global logger.
+Normally this would be the ConsoleLogger in the REPL etc.
+"""
+function demux_global_logger!(loggers...; include_current=true)
+	loggers = collect(loggers)
+	if include_current
+		push!(loggers, global_logger())
+	end
+	global_logger(DemuxLogger(loggers))
+end
+
+function handle_message(demux::demux, level, message, _module, group, id, file, line; kwargs...)
+	for logger in demux.loggers
+		if shouldlog(logger,  level, _module, group, id)
+			handle_message(logger, level, message, _module, group, id, file, line; kwargs...)
+		end
+	end
+end
+
+function shouldlog(demux::demux, level, _module, group, id)
+	any(shouldlog(logger, level, _module, group, id) for logger in demux.loggers)
+end
+
+function min_enabled_level(demux::demux)
+	minimum(min_enabled_level(logger) for logger in demux.loggers)
+end
+
+function catch_exceptions(demux::demux)
+	any(catch_exceptions(logger) for logger in demux.loggers)
+end

src/filelogger.jl

@@ -0,0 +1,18 @@
+struct FileLogger <: AbstractLogger
+	logger::SimpleLogger
+	function FileLogger(path, level=Info; append=false)
+		filehandle = open(path, append ? "a" : "w")
+		this = new(SimpleLogger(filehandle, level))
+		finalizer(this) do this
+			close(filehandle)
+		end
+	end
+end
+
+
+function handle_message(filelogger::FileLogger, level, message, _module, group, id, file, line; kwargs...)
+	handle_message(filelogger.logger, level, message, _module, group, id, file, line; kwargs...)
+end
+shouldlog(filelogger::FileLogger, level, _module, group, id) = shouldlog(filelogger.logger, level, _module, group, id)
+min_enabled_level(filelogger::FileLogger) = min_enabled_level(filelogger.logger)
+catch_exceptions(filelogger::FileLogger) = catch_exceptions(filelogger.logger)

src/filteredlogger.jl

@@ -0,0 +1,44 @@
+
+"""
+	FilteredLogger(filter, logger)
+	
+Wraps `logger` in a filter.
+While loggers intrinsictally have in built filtering mechanisms.
+Wrapping it in a FilterLogger allows for extract control,
+at the cost of a bit of overhead.
+
+The `filter` should be a function that returns a boolean.
+`true` if the message should be logged and `false` if not.
+It should take as inputs:
+`level, message, _module, group, id, file, line; kwargs...)`
+
+These argument come from the logging macro (@info`, `@warn` etc).
+  
+  * `level::LogLevel` Warn, Info, etc,
+  * `message::String` the message to be logged
+  * `_module::Module` can be used to specify a different originating module from
+    the source location of the message.
+  * `group::Symbol` can be used to override the message group (this is
+    normally derived from the base name of the source file).
+  * `id::Symbol` can be used to override the automatically generated unique
+    message identifier.  This is useful if you need to very closely associate
+    messages generated on different source lines.
+  * `file::String` and `line::Int` can be used to override the apparent
+    source location of a log message.
+  * `kwargs...` any keyword or positional arguments to the logging macro.
+"""
+struct FilteredLogger{T,F} <: AbstractLogger where T<: AbstractLogger
+	filter::F
+	logger::SimpleLogger
+end
+
+
+function handle_message(filteredlogger::FilteredLogger, level, message, _module, group, id, file, line; kwargs...)
+	if filteredlogger.filter(filteredlogger.logger, level, message, _module, group, id, file, line; kwargs...)
+		handle_message(filteredlogger.logger, level, message, _module, group, id, file, line; kwargs...)
+	end
+end
+
+shouldlog(filteredlogger::FilteredLogger, level, _module, group, id) = shouldlog(filteredlogger.logger, level, _module, group, id)
+min_enabled_level(filteredlogger::FilteredLogger) = min_enabled_level(filteredlogger.logger)
+catch_exceptions(filteredlogger::FilteredLogger) = catch_exceptions(filteredlogger.logger)