cyfs_debug/dump/
helper.rs

1use crate::DebugConfig;
2use cyfs_base::*;
3
4use std::ffi::OsStr;
5use std::path::{Path, PathBuf};
6
7pub struct DumpHelper {
8    enable_dump: bool,
9    full_dump: bool,
10}
11
12impl DumpHelper {
13    fn new() -> Self {
14        let enable_dump = match get_channel() {
15            CyfsChannel::Nightly => true,
16            _ => false,
17        };
18
19        Self {
20            enable_dump,
21            full_dump: false,
22        }
23    }
24
25    pub fn get_instance() -> &'static Self {
26        use once_cell::sync::OnceCell;
27        static S_INSTANCE: OnceCell<DumpHelper> = OnceCell::new();
28        S_INSTANCE.get_or_init(|| {
29            let mut ret = Self::new();
30            ret.load_config();
31            ret
32        })
33    }
34
35    pub fn is_enable_dump(&self) -> bool {
36        self.enable_dump
37    }
38
39    fn default_basename() -> String {
40        let arg0 = std::env::args().next().unwrap_or_else(|| "cyfs".to_owned());
41        Path::new(&arg0).file_stem().map(OsStr::to_string_lossy).unwrap(/*cannot fail*/).to_string()
42    }
43
44    fn dump_file_name() -> String {
45        let id = std::process::id();
46        let now = chrono::Local::now();
47        let now = now.format("%Y-%m-%d_%H-%M-%S%.6f_%z");
48
49        format!("{}_{}_{}.dmp", Self::default_basename(), id, now)
50    }
51
52    fn dump_dir() -> PathBuf {
53        let dump_dir = cyfs_util::get_log_dir("core-dump");
54        if !dump_dir.is_dir() {
55            if let Err(e) = std::fs::create_dir_all(&dump_dir) {
56                error!(
57                    "create core-dump dir failed! dir={}, err={}",
58                    dump_dir.display(),
59                    e
60                );
61            }
62        }
63
64        dump_dir
65    }
66
67    fn load_config(&mut self) {
68        if let Some(config_node) = DebugConfig::get_config("dump") {
69            if let Err(e) = self.load_config_value(config_node) {
70                println!("load process dead check config error! {}", e);
71            }
72        }
73    }
74
75    fn load_config_value(&mut self, config_node: &toml::Value) -> BuckyResult<()> {
76        let node = config_node.as_table().ok_or_else(|| {
77            let msg = format!("invalid dump config format! content={}", config_node,);
78            error!("{}", msg);
79
80            BuckyError::new(BuckyErrorCode::InvalidFormat, msg)
81        })?;
82
83        for (k, v) in node {
84            match k.as_str() {
85                "enable" => {
86                    if let Some(v) = v.as_bool() {
87                        println!(
88                            "load dump.enable from config: {}, current={}",
89                            v, self.enable_dump
90                        );
91                        self.enable_dump = v;
92                    } else {
93                        println!("unknown dump.enable config node: {:?}", v);
94                    }
95                }
96                "full" => {
97                    if let Some(v) = v.as_bool() {
98                        println!(
99                            "load dump.full from config: {}, current={}",
100                            v, self.full_dump
101                        );
102                        self.enable_dump = v;
103                    } else {
104                        println!("unknown dump.full config node: {:?}", v);
105                    }
106                }
107
108                key @ _ => {
109                    println!("unknown dump config node: {}={:?}", key, v);
110                }
111            }
112        }
113
114        Ok(())
115    }
116
117    pub fn dump(&self) {
118        let dir = Self::dump_dir();
119        let filename = Self::dump_file_name();
120
121        info!("will create dump file: {}/{}", dir.display(), filename);
122
123        super::dump::create_dump(&dir, &filename, self.full_dump)
124    }
125}
cyfs_debug/dump/helper.rs

cyfs_debug/dump/
helper.rs
