fix: fix test_scheduler_continuous_stop in scheduler (GreptimeTeam#2252)

* fix: fix test_scheduler_continuous_stop in scheduler Signed-off-by: ZhuZiyi <[email protected]> * chore: add document annotation Signed-off-by: ZhuZiyi <[email protected]> --------- Signed-off-by: ZhuZiyi <[email protected]>
paomian · Oct 19, 2023 · a377bc8 · a377bc8
1 parent 49c298d
commit a377bc8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/mito2/src/schedule/scheduler.rs b/src/mito2/src/schedule/scheduler.rs
@@ -125,6 +125,7 @@ impl Scheduler for LocalScheduler {
             .map_err(|_| InvalidFlumeSenderSnafu {}.build())
     }
 
+    /// if await_termination is true, scheduler will wait all tasks finished before stopping
     async fn stop(&self, await_termination: bool) -> Result<()> {
         ensure!(
             self.state.load(Ordering::Relaxed) == STATE_RUNNING,
@@ -235,7 +236,7 @@ mod tests {
         let local_stop = local.clone();
         tokio::spawn(async move {
             tokio::time::sleep(Duration::from_millis(5)).await;
-            local_stop.stop(false).await.unwrap();
+            local_stop.stop(true).await.unwrap();
             barrier_clone.wait().await;
         });