fix: reduce unsafe unwrap

This commit is contained in:
GyDi 2022-08-23 15:19:04 +08:00 committed by GitHub
parent 30f9f1a021
commit 3242efb1a2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,10 +1,9 @@
use super::Core;
use crate::log_if_err;
use crate::utils::help::get_now;
use anyhow::{bail, Context, Result};
use delay_timer::prelude::{DelayTimer, DelayTimerBuilder, TaskBuilder};
use std::collections::HashMap;
use std::ops::Mul;
use std::time::{SystemTime, UNIX_EPOCH};
type TaskID = u64;
@ -67,33 +66,29 @@ impl Timer {
/// restore timer
pub fn restore(&mut self) -> Result<()> {
log_if_err!(self.refresh());
let profiles = self.core.as_ref().unwrap().profiles.lock();
let cur_timestamp = SystemTime::now()
.duration_since(UNIX_EPOCH)
.unwrap()
.as_secs() as usize;
self.refresh()?;
let cur_timestamp = get_now(); // seconds
let profiles = self.core.as_ref().unwrap().profiles.lock();
profiles
.get_items()
.unwrap_or(&vec![])
.iter()
.filter(|item| item.uid.is_some() && item.updated.is_some() && item.option.is_some())
.filter(|item| {
// mins to seconds
let interval = item.option.as_ref().unwrap().update_interval.unwrap_or(0) as usize * 60;
let updated = item.updated.unwrap();
return interval > 0 && cur_timestamp - updated >= interval;
})
.for_each(|item| {
let uid = item.uid.as_ref().unwrap();
if let Some((task_id, _)) = self.timer_map.get(uid) {
log_if_err!(self.delay_timer.advance_task(*task_id));
}
});
for item in profiles.get_items().unwrap() {
// if current_time - last_update_time >= interval, cron job should execute immediately.
if cur_timestamp - item.updated.unwrap()
>= item
.option
.as_ref()
.unwrap()
.update_interval
.unwrap_or(0xffffffff)
.mul(60) // minute to secs
.try_into()
.unwrap()
{
let (task_id, _) = self
.timer_map
.get(&item.uid.as_ref().unwrap().clone())
.unwrap();
log_if_err!(self.delay_timer.advance_task(*task_id));
}
}
Ok(())
}