fix

Signed-off-by: andylokandy <[email protected]>
tikv · Jun 22, 2024 · 9a4a1df · 9a4a1df
1 parent 2b9438b
commit 9a4a1df
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/minitrace/src/collector/global_collector.rs b/minitrace/src/collector/global_collector.rs
@@ -99,11 +99,9 @@ pub fn flush() {
             std::thread::Builder::new()
                 .name("minitrace-flush".to_string())
                 .spawn(move || {
-                    GLOBAL_COLLECTOR
-                        .lock()
-                        .as_mut()
-                        .unwrap()
-                        .handle_commands(true);
+                    if let Some(global_collector) = GLOBAL_COLLECTOR.lock().as_mut() {
+                        global_collector.handle_commands(true);
+                    }
                 })
                 .unwrap()
                 .join()
@@ -234,9 +232,11 @@ impl GlobalCollector {
                 .spawn(move || {
                     loop {
                         let begin_instant = Instant::now();
-                        if let Some(global_collector) = GLOBAL_COLLECTOR.lock().as_mut() {
-                            global_collector.handle_commands(false);
-                        }
+                        GLOBAL_COLLECTOR
+                            .lock()
+                            .as_mut()
+                            .unwrap()
+                            .handle_commands(false);
                         std::thread::sleep(
                             config
                                 .background_collect_interval