rust/sat-rs
rust/sat-rs
5
1
Fork 0
Code Issues 9 Pull Requests Projects Releases 22 Wiki Activity

closure returns a boolean whether to del or not

Browse Source
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
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
satrs-core/src/pus/scheduling.rs
View File

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