closure returns a boolean whether to del or not

This commit is contained in:
Robin Müller 2023-01-25 10:43:11 +01:00
parent 19d43b1b2c
commit 36e63bc9a9
No known key found for this signature in database
GPG Key ID: 71B58F8A3CDFA9AC

View File

@ -126,24 +126,33 @@ impl PusScheduler {
/// Utility method which calls [Self::telecommands_to_release] and then calls a releaser
/// closure for each telecommand which should be released. This function will also delete
/// the telecommands from the holding store after calling the release closure.
pub fn release_telecommands<R: FnMut(bool, &StoreAddr)>(
///
/// # Arguments
///
/// * `releaser` - Closure where the first argument is whether the scheduler is enabled and
/// the second argument is the store address. This closure should return whether the
/// command should be deleted.
/// * `store` - The holding store of the telecommands.
pub fn release_telecommands<R: FnMut(bool, &StoreAddr) -> bool>(
&mut self,
mut releaser: R,
store: &mut impl PoolProvider,
tc_store: &mut impl PoolProvider,
) -> Result<u64, (u64, StoreError)> {
let tcs_to_release = self.telecommands_to_release();
let mut released_tcs = 0;
let mut store_error = Ok(());
for tc in tcs_to_release {
for addr in tc.1 {
releaser(self.enabled, addr);
let should_delete = releaser(self.enabled, addr);
released_tcs += 1;
let res = store.delete(*addr);
if should_delete {
let res = tc_store.delete(*addr);
if res.is_err() {
store_error = res;
}
}
}
}
self.tc_map.retain(|k, _| k > &self.current_time);
store_error
.map(|_| released_tcs)
@ -273,6 +282,7 @@ mod tests {
let mut i = 0;
let mut test_closure_1 = |boolvar: bool, store_addr: &StoreAddr| {
common_check(boolvar, store_addr, vec![first_addr], &mut i);
true
};
// test 1: too early, no tcs
@ -294,6 +304,7 @@ mod tests {
// test 3, late timestamp, release 1 overdue tc
let mut test_closure_2 = |boolvar: bool, store_addr: &StoreAddr| {
common_check(boolvar, store_addr, vec![second_addr], &mut i);
true
};
scheduler.update_time(UnixTimestamp::new_only_seconds(206));
@ -328,6 +339,7 @@ mod tests {
let mut i = 0;
let mut test_closure = |boolvar: bool, store_addr: &StoreAddr| {
common_check(boolvar, store_addr, vec![first_addr, second_addr], &mut i);
true
};
// test 1: too early, no tcs