pdk_core/log/
logger.rs

1// Copyright (c) 2025, Salesforce, Inc.,
2// All rights reserved.
3// For full license text, see the LICENSE.txt file
4
5//! Function to manipulate the log level.
6
7use 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
21/// Configures the logger to use.
22pub 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
52/// Translates proxy_wasm log level to the log crate level.
53pub 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
64/// Translates the log crate level to the proxy_wasm log level.
65pub 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}