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

Avoid nested loops

Browse Source
This commit is contained in:
Ruihan Li
2024-09-01 18:54:17 +08:00
committed by Tate, Hongliang Tian
parent 25daab7e78
commit d70ae181b5
1 changed files with 35 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
kernel/src/fs/epoll/epoll_file.rs
View File

@ -212,28 +212,20 @@ impl EpollFile {
} }
fn pop_ready(&self, max_events: usize, ep_events: &mut Vec<EpollEvent>) -> usize { fn pop_ready(&self, max_events: usize, ep_events: &mut Vec<EpollEvent>) -> usize {
let mut count_events = 0;
let mut ready = self.ready.lock(); let mut ready = self.ready.lock();
let mut pop_quota = ready.len();
loop { let mut count_events = 0;
// Pop some ready entries per round. for _ in 0..ready.len() {
// if count_events >= max_events {
// Since the popped ready entries may contain "false positive" and
// we want to return as many results as possible, this has to
// be done in a loop.
let pop_count = (max_events - count_events).min(pop_quota);
if pop_count == 0 {
break; break;
} }
let ready_entries: Vec<Arc<EpollEntry>> = ready
.drain(..pop_count)
.filter_map(|entry| Weak::upgrade(&entry))
.collect();
pop_quota -= pop_count;
// Examine these ready entries, which are candidates for the results let weak_entry = ready.pop_front().unwrap();
// to be returned. let Some(entry) = Weak::upgrade(&weak_entry) else {
for entry in ready_entries { // The entry has been deleted.
continue;
};
let (ep_event, ep_flags) = entry.event_and_flags(); let (ep_event, ep_flags) = entry.event_and_flags();
// If this entry's file is ready, save it in the output array. // If this entry's file is ready, save it in the output array.
// EPOLLHUP and EPOLLERR should always be reported. // EPOLLHUP and EPOLLERR should always be reported.
@ -250,7 +242,7 @@ impl EpollFile {
if !ready_events.is_empty() if !ready_events.is_empty()
&& !ep_flags.intersects(EpollFlags::ONE_SHOT | EpollFlags::EDGE_TRIGGER) && !ep_flags.intersects(EpollFlags::ONE_SHOT | EpollFlags::EDGE_TRIGGER)
{ {
ready.push_back(Arc::downgrade(&entry)); ready.push_back(weak_entry);
} }
// Otherwise, the entry is indeed removed the ready list and we should reset // Otherwise, the entry is indeed removed the ready list and we should reset
// its ready flag. // its ready flag.
@ -263,12 +255,12 @@ impl EpollFile {
} }
} }
} }
}
// Clear the epoll file's events if no ready entries // Clear the epoll file's events if no ready entries
if ready.len() == 0 { if ready.len() == 0 {
self.pollee.del_events(IoEvents::IN); self.pollee.del_events(IoEvents::IN);
} }
count_events count_events
} }