Samuka007/asterinas
1
0
Fork 0
mirror of https://github.com/asterinas/asterinas.git synced 2025-06-09 13:26:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add schedule after WorkItem enqueue

Browse Source
This commit is contained in:
Yuke Peng 2024-06-08 16:17:00 +08:00 committed by Tate, Hongliang Tian
parent 4afac4d5ac
commit 89ce1f4df9
3 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
kernel/aster-nix/src/thread/work_queue/mod.rs
View File

@ -141,6 +141,9 @@ impl WorkQueue {
.lock_irq_disabled() .lock_irq_disabled()
.pending_work_items .pending_work_items
.push(work_item); .push(work_item);
if let Some(worker_pool) = self.worker_pool.upgrade() {
worker_pool.schedule()
}
true true
} }

4
kernel/aster-nix/src/thread/work_queue/worker.rs
View File

@ -97,6 +97,10 @@ impl Worker {
self.exit(); self.exit();
} }
pub(super) fn bound_thread(&self) -> &Arc<Thread> {
&self.bound_thread
}
pub(super) fn is_idle(&self) -> bool { pub(super) fn is_idle(&self) -> bool {
self.inner.lock_irq_disabled().worker_status == WorkerStatus::Idle self.inner.lock_irq_disabled().worker_status == WorkerStatus::Idle
} }

13
kernel/aster-nix/src/thread/work_queue/worker_pool.rs
View File

@ -2,9 +2,16 @@
#![allow(dead_code)] #![allow(dead_code)]
use core::sync::atomic::{AtomicBool, Ordering}; use core::{
sync::atomic::{AtomicBool, Ordering},
time::Duration,
};
use aster_frame::{cpu::CpuSet, sync::WaitQueue, task::Priority}; use aster_frame::{
cpu::CpuSet,
sync::WaitQueue,
task::{add_task, Priority},
};
use super::{simple_scheduler::SimpleScheduler, worker::Worker, WorkItem, WorkPriority, WorkQueue}; use super::{simple_scheduler::SimpleScheduler, worker::Worker, WorkItem, WorkPriority, WorkQueue};
use crate::{ use crate::{
@ -74,7 +81,7 @@ impl LocalWorkerPool {
fn add_worker(&self) { fn add_worker(&self) {
let worker = Worker::new(self.parent.clone(), self.cpu_id); let worker = Worker::new(self.parent.clone(), self.cpu_id);
self.workers.lock_irq_disabled().push_back(worker.clone()); self.workers.lock_irq_disabled().push_back(worker.clone());
worker.run(); add_task(worker.bound_thread().task().clone());
} }
fn remove_worker(&self) { fn remove_worker(&self) {