Add num tasks in scheduler

Author: hky1999

src/cfs.rs

@@ -187,4 +187,8 @@ impl<T> BaseScheduler for CFScheduler<T> {
             false
         }
     }
+
+    fn num_tasks(&self) -> usize {
+        self.ready_queue.len()
+    }
 }

src/fifo.rs

@@ -56,13 +56,15 @@ impl<T> Deref for FifoTask<T> {
 /// It internally uses a linked list as the ready queue.
 pub struct FifoScheduler<T> {
     ready_queue: List<Arc<FifoTask<T>>>,
+    num_tasks: usize,
 }
 
 impl<T> FifoScheduler<T> {
     /// Creates a new empty [`FifoScheduler`].
     pub const fn new() -> Self {
         Self {
             ready_queue: List::new(),
+            num_tasks: 0,
         }
     }
     /// get the name of scheduler
@@ -78,17 +80,21 @@ impl<T> BaseScheduler for FifoScheduler<T> {
 
     fn add_task(&mut self, task: Self::SchedItem) {
         self.ready_queue.push_back(task);
+        self.num_tasks += 1;
     }
 
     fn remove_task(&mut self, task: &Self::SchedItem) -> Option<Self::SchedItem> {
+        self.num_tasks -= 1;
         unsafe { self.ready_queue.remove(task) }
     }
 
     fn pick_next_task(&mut self) -> Option<Self::SchedItem> {
+        self.num_tasks -= 1;
         self.ready_queue.pop_front()
     }
 
     fn put_prev_task(&mut self, prev: Self::SchedItem, _preempt: bool) {
+        self.num_tasks += 1;
         self.ready_queue.push_back(prev);
     }
 
@@ -99,4 +105,8 @@ impl<T> BaseScheduler for FifoScheduler<T> {
     fn set_priority(&mut self, _task: &Self::SchedItem, _prio: isize) -> bool {
         false
     }
+
+    fn num_tasks(&self) -> usize {
+        self.num_tasks
+    }
 }

src/lib.rs

@@ -63,6 +63,9 @@ pub trait BaseScheduler {
     /// `current` is the current running task.
     fn task_tick(&mut self, current: &Self::SchedItem) -> bool;
 
-    /// set priority for a task
+    /// Set priority for a task.
     fn set_priority(&mut self, task: &Self::SchedItem, prio: isize) -> bool;
+
+    /// Returns the number of tasks in the scheduler.
+    fn num_tasks(&self) -> usize;
 }

src/round_robin.rs

@@ -110,4 +110,8 @@ impl<T, const S: usize> BaseScheduler for RRScheduler<T, S> {
     fn set_priority(&mut self, _task: &Self::SchedItem, _prio: isize) -> bool {
         false
     }
+
+    fn num_tasks(&self) -> usize {
+        self.ready_queue.len()
+    }
 }