os/task/
manager.rs

1use super::{ProcessControlBlock, TaskControlBlock, TaskStatus};
2use crate::sync::UPSafeCell;
3use alloc::collections::{BTreeMap, VecDeque};
4use alloc::sync::Arc;
5use lazy_static::*;
6
7pub struct TaskManager {
8    ready_queue: VecDeque<Arc<TaskControlBlock>>,
9}
10
11/// A simple FIFO scheduler.
12impl TaskManager {
13    pub fn new() -> Self {
14        Self {
15            ready_queue: VecDeque::new(),
16        }
17    }
18    pub fn add(&mut self, task: Arc<TaskControlBlock>) {
19        self.ready_queue.push_back(task);
20    }
21    pub fn fetch(&mut self) -> Option<Arc<TaskControlBlock>> {
22        self.ready_queue.pop_front()
23    }
24    pub fn remove(&mut self, task: Arc<TaskControlBlock>) {
25        if let Some((id, _)) = self
26            .ready_queue
27            .iter()
28            .enumerate()
29            .find(|(_, t)| Arc::as_ptr(t) == Arc::as_ptr(&task))
30        {
31            self.ready_queue.remove(id);
32        }
33    }
34}
35
36lazy_static! {
37    pub static ref TASK_MANAGER: UPSafeCell<TaskManager> =
38        unsafe { UPSafeCell::new(TaskManager::new()) };
39    pub static ref PID2PCB: UPSafeCell<BTreeMap<usize, Arc<ProcessControlBlock>>> =
40        unsafe { UPSafeCell::new(BTreeMap::new()) };
41}
42
43pub fn add_task(task: Arc<TaskControlBlock>) {
44    TASK_MANAGER.exclusive_access().add(task);
45}
46
47pub fn wakeup_task(task: Arc<TaskControlBlock>) {
48    let mut task_inner = task.inner_exclusive_access();
49    task_inner.task_status = TaskStatus::Ready;
50    drop(task_inner);
51    add_task(task);
52}
53
54pub fn remove_task(task: Arc<TaskControlBlock>) {
55    TASK_MANAGER.exclusive_access().remove(task);
56}
57
58pub fn fetch_task() -> Option<Arc<TaskControlBlock>> {
59    TASK_MANAGER.exclusive_access().fetch()
60}
61
62pub fn pid2process(pid: usize) -> Option<Arc<ProcessControlBlock>> {
63    let map = PID2PCB.exclusive_access();
64    map.get(&pid).map(Arc::clone)
65}
66
67pub fn insert_into_pid2process(pid: usize, process: Arc<ProcessControlBlock>) {
68    PID2PCB.exclusive_access().insert(pid, process);
69}
70
71pub fn remove_from_pid2process(pid: usize) {
72    let mut map = PID2PCB.exclusive_access();
73    if map.remove(&pid).is_none() {
74        panic!("cannot find pid {} in pid2task!", pid);
75    }
76}
os/task/manager.rs

os/task/
manager.rs
