stalwartlabs/mail-server
1
1
Fork 0
mirror of https://github.com/stalwartlabs/mail-server.git synced 2025-11-08 12:51:07 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove from locked task queue events no longer present in the db

Browse source
This commit is contained in:
mdecimus 2025-06-07 23:11:22 +02:00
parent fe074ad369
commit 10e932a43a
1 changed files with 19 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
crates/services/src/task_manager/mod.rs
View file

@ -61,7 +61,13 @@ pub(crate) struct TaskManagerIpc {
tx_fts: mpsc::Sender<Task>,
tx_bayes: mpsc::Sender<Task>,
tx_alarm: mpsc::Sender<Task>,
locked: AHashMap<Vec<u8>, Instant>,
locked: AHashMap<Vec<u8>, Locked>,
revision: u64,
}
struct Locked {
expires: Instant,
revision: u64,
}
pub fn spawn_task_manager(inner: Arc<Inner>) {
@ -144,6 +150,7 @@ pub fn spawn_task_manager(inner: Arc<Inner>) {
tx_bayes: tx_index_2,
tx_alarm: tx_index_3,
locked: Default::default(),
revision: 0,
};
let rx = inner.ipc.task_tx.clone();
loop {
@ -188,6 +195,7 @@ impl TaskQueueManager for Server {
let now_timestamp = now();
let now = Instant::now();
let mut next_event = None;
ipc.revision += 1;
let _ = self
.core
.storage
@ -199,18 +207,20 @@ impl TaskQueueManager for Server {
if task.due <= now_timestamp {
match ipc.locked.entry(key.to_vec()) {
Entry::Occupied(mut entry) => {
let expires = entry.get_mut();
if *expires <= now {
*expires = Instant::now()
let locked = entry.get_mut();
if locked.expires <= now {
locked.expires = Instant::now()
+ std::time::Duration::from_secs(task.lock_expiry() + 1);
tasks.push(task);
}
locked.revision = ipc.revision;
}
Entry::Vacant(entry) => {
entry.insert(
Instant::now()
entry.insert(Locked {
expires: Instant::now()
+ std::time::Duration::from_secs(task.lock_expiry() + 1),
);
revision: ipc.revision,
});
tasks.push(task);
}
}
@ -260,7 +270,8 @@ impl TaskQueueManager for Server {
// Delete expired locks
let now = Instant::now();
ipc.locked.retain(|_, expires| *expires > now);
ipc.locked
.retain(|_, locked| locked.expires > now && locked.revision == ipc.revision);
next_event.map_or(LONG_1Y_SLUMBER, |timestamp| {
Duration::from_secs(timestamp.saturating_sub(store::write::now()))
})