Merge pull request #303 from apecloud/fix/fix_mysql_structure_fetcher

fix stack backtrace missing when panic happens
apecloud · Dec 24, 2024 · 0937334 · 0937334
2 parents b6c7048 + 3219230
commit 0937334
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dt-main/src/main.rs b/dt-main/src/main.rs
@@ -1,10 +1,12 @@
-use std::env;
+use std::{env, panic};
 
 use dt_precheck::{config::task_config::PrecheckTaskConfig, do_precheck};
 use dt_task::task_runner::TaskRunner;
 
 #[tokio::main]
 async fn main() {
+    env::set_var("RUST_BACKTRACE", "1");
+
     let args: Vec<String> = env::args().collect();
     if args.len() < 2 {
         panic!("no task_config provided in args");

diff --git a/dt-task/src/task_runner.rs b/dt-task/src/task_runner.rs
@@ -1,5 +1,4 @@
 use std::{
-    env,
     fs::{self, File},
     io::Read,
     panic,
@@ -74,7 +73,6 @@ impl TaskRunner {
             self.init_log4rs()?;
         }
 
-        env::set_var("RUST_BACKTRACE", "1");
         panic::set_hook(Box::new(|panic_info| {
             log_error!("panic: {}", panic_info);
         }));