refactor: pop tasks before whole runtime shutdown (#208)

bytedance · Aug 31, 2023 · 5b9c503 · 5b9c503
1 parent 716a9a3
commit 5b9c503
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/monoio/src/scheduler.rs b/monoio/src/scheduler.rs
@@ -27,6 +27,15 @@ impl Default for TaskQueue {
     }
 }
 
+impl Drop for TaskQueue {
+    fn drop(&mut self) {
+        unsafe {
+            let queue = &mut *self.queue.get();
+            while let Some(_task) = queue.pop_front() {}
+        }
+    }
+}
+
 impl TaskQueue {
     pub(crate) fn new() -> Self {
         const DEFAULT_TASK_QUEUE_SIZE: usize = 4096;