[provider] set the stop signal earlier

liuchengxu · May 2, 2024 · e939219 · e939219
1 parent 4eafae2
commit e939219
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crates/maple_core/src/stdio_server/provider/mod.rs b/crates/maple_core/src/stdio_server/provider/mod.rs
@@ -82,8 +82,8 @@ struct SearcherControl {
 
 impl SearcherControl {
     fn kill_in_background(self) {
+        self.stop_signal.store(true, Ordering::SeqCst);
         if !self.join_handle.is_finished() {
-            self.stop_signal.store(true, Ordering::SeqCst);
             // NOTE: The kill operation may take some time, using `spawn_block` to not block current thread.
             tokio::task::spawn_blocking(move || {
                 self.join_handle.abort();