1use std::panic;
8use std::sync::atomic::{AtomicBool, Ordering};
9
10use crate::HostTrait;
11use classy::proxy_wasm::types::LogLevel;
12
13use crate::host::property::PropertyAccessor;
14use crate::log::log_metadata::LogMetadata;
15
16struct Logger;
17
18static LOGGER: Logger = Logger;
19static INITIALIZED: AtomicBool = AtomicBool::new(false);
20
21pub fn set_log_level(level: LogLevel) {
23 if !INITIALIZED.load(Ordering::Relaxed) {
24 let _ = log::set_logger(&LOGGER);
25 panic::set_hook(Box::new(|panic_info| {
26 crate::Host.log(LogLevel::Critical, &panic_info.to_string());
27 }));
28 INITIALIZED.store(true, Ordering::Relaxed);
29 }
30 log::set_max_level(to_log_lib_level(level));
31}
32
33impl log::Log for Logger {
34 fn enabled(&self, metadata: &log::Metadata) -> bool {
35 metadata.level() <= log::max_level()
36 }
37
38 fn log(&self, record: &log::Record) {
39 if !self.enabled(record.metadata()) {
40 return;
41 }
42
43 let metadata = LogMetadata::from(<dyn PropertyAccessor>::default());
44 let message = format!("{} {}", metadata, &record.args());
45 let level = to_proxy_level(record.level());
46 crate::Host.log(level, &message);
47 }
48
49 fn flush(&self) {}
50}
51
52pub fn to_log_lib_level(level: LogLevel) -> log::LevelFilter {
54 match level {
55 LogLevel::Trace => log::LevelFilter::Trace,
56 LogLevel::Debug => log::LevelFilter::Debug,
57 LogLevel::Info => log::LevelFilter::Info,
58 LogLevel::Warn => log::LevelFilter::Warn,
59 LogLevel::Error => log::LevelFilter::Error,
60 LogLevel::Critical => log::LevelFilter::Off,
61 }
62}
63
64pub fn to_proxy_level(level: log::Level) -> LogLevel {
66 match level {
67 log::Level::Trace => LogLevel::Trace,
68 log::Level::Debug => LogLevel::Debug,
69 log::Level::Info => LogLevel::Info,
70 log::Level::Warn => LogLevel::Warn,
71 log::Level::Error => LogLevel::Error,
72 }
73}