burn 0.3.0

BURN: Burn Unstoppable Rusty Neurons
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

use log::LevelFilter;
use log4rs::{
    append::file::FileAppender,
    config::{Appender, Config, Root},
    encode::pattern::PatternEncoder,
    Handle,
};
use std::{cell::Cell, sync::Mutex};

static LOGGER_HANDLER: Mutex<Cell<Option<Handle>>> = Mutex::new(Cell::new(None));

pub fn update_log_file(file_path: &str) {
    let config = create_config(file_path);
    set_config(config, file_path);
}

fn create_config(file_path: &str) -> Config {
    let experiment = FileAppender::builder()
        .encoder(Box::new(PatternEncoder::new(
            "[{d(%+)(utc)} - {h({l})} - {f}:{L}] {m}{n}",
        )))
        .build(file_path)
        .unwrap();

    Config::builder()
        .appender(Appender::builder().build("experiment", Box::new(experiment)))
        .build(
            Root::builder()
                .appender("experiment")
                .build(LevelFilter::Info),
        )
        .unwrap()
}

fn set_config(config: Config, file_path: &str) {
    let mut cell = LOGGER_HANDLER.lock().unwrap();

    match cell.get_mut() {
        Some(handler) => {
            handler.set_config(config);
        }
        None => {
            let handler = log4rs::init_config(config).unwrap();
            update_panic_hook(file_path);
            cell.replace(Some(handler));
        }
    }
}

fn update_panic_hook(file_path: &str) {
    let hook = std::panic::take_hook();
    let file_path = file_path.to_owned();

    std::panic::set_hook(Box::new(move |info| {
        log::error!("PANIC => {}", info.to_string());
        eprintln!(
            "=== PANIC ===\nA fatal error happened, you can check the experiment logs here => '{}'\n=============",
            file_path
        );
        hook(info);
    }));
}