Samuka007/asterinas
1
0
Fork 0
mirror of https://github.com/asterinas/asterinas.git synced 2025-06-23 17:33:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Sets the default scheduler in aster_frame

Browse Source
This commit is contained in:
Jianfeng Jiang
2024-06-20 08:13:58 +00:00
committed by Tate, Hongliang Tian
parent 18cc7f0a6b
commit ef075d15d0
1 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
framework/aster-frame/src/task/scheduler.rs
View File

@ -4,14 +4,12 @@
use alloc::collections::VecDeque; use alloc::collections::VecDeque;
use lazy_static::lazy_static;
use crate::{prelude::*, sync::SpinLock, task::Task}; use crate::{prelude::*, sync::SpinLock, task::Task};
lazy_static! { static DEFAULT_SCHEDULER: FifoScheduler = FifoScheduler::new();
pub(crate) static ref GLOBAL_SCHEDULER: SpinLock<GlobalScheduler> = pub(crate) static GLOBAL_SCHEDULER: SpinLock<GlobalScheduler> = SpinLock::new(GlobalScheduler {
SpinLock::new(GlobalScheduler { scheduler: None }); scheduler: &DEFAULT_SCHEDULER,
} });
/// A scheduler for tasks. /// A scheduler for tasks.
/// ///
@ -29,34 +27,37 @@ pub trait Scheduler: Sync + Send {
} }
pub struct GlobalScheduler { pub struct GlobalScheduler {
scheduler: Option<&'static dyn Scheduler>, scheduler: &'static dyn Scheduler,
} }
impl GlobalScheduler { impl GlobalScheduler {
pub fn new() -> Self { pub const fn new(scheduler: &'static dyn Scheduler) -> Self {
Self { scheduler: None } Self { scheduler }
} }
/// dequeue a task using scheduler /// dequeue a task using scheduler
/// require the scheduler is not none /// require the scheduler is not none
pub fn dequeue(&mut self) -> Option<Arc<Task>> { pub fn dequeue(&mut self) -> Option<Arc<Task>> {
self.scheduler.unwrap().dequeue() self.scheduler.dequeue()
} }
/// enqueue a task using scheduler /// enqueue a task using scheduler
/// require the scheduler is not none /// require the scheduler is not none
pub fn enqueue(&mut self, task: Arc<Task>) { pub fn enqueue(&mut self, task: Arc<Task>) {
self.scheduler.unwrap().enqueue(task) self.scheduler.enqueue(task)
} }
pub fn should_preempt(&self, task: &Arc<Task>) -> bool { pub fn should_preempt(&self, task: &Arc<Task>) -> bool {
self.scheduler.unwrap().should_preempt(task) self.scheduler.should_preempt(task)
} }
} }
/// Set the global task scheduler. /// Sets the global task scheduler.
/// ///
/// This must be called before invoking `Task::spawn`. /// This must be called before invoking `Task::spawn`.
pub fn set_scheduler(scheduler: &'static dyn Scheduler) { pub fn set_scheduler(scheduler: &'static dyn Scheduler) {
GLOBAL_SCHEDULER.lock_irq_disabled().scheduler = Some(scheduler); let mut global_scheduler = GLOBAL_SCHEDULER.lock_irq_disabled();
// When setting a new scheduler, the old scheduler should be empty
assert!(global_scheduler.dequeue().is_none());
global_scheduler.scheduler = scheduler;
} }
pub fn fetch_task() -> Option<Arc<Task>> { pub fn fetch_task() -> Option<Arc<Task>> {