use {
super::{SnapshotKind, SnapshotPackage},
std::cmp::Ordering::{self, Equal, Greater, Less},
};
#[must_use]
pub fn cmp_snapshot_packages_by_priority(a: &SnapshotPackage, b: &SnapshotPackage) -> Ordering {
cmp_snapshot_kinds_by_priority(&a.snapshot_kind, &b.snapshot_kind).then(a.slot.cmp(&b.slot))
}
#[must_use]
pub fn cmp_snapshot_kinds_by_priority(a: &SnapshotKind, b: &SnapshotKind) -> Ordering {
use SnapshotKind as Kind;
match (a, b) {
(Kind::FullSnapshot, Kind::FullSnapshot) => Equal,
(Kind::FullSnapshot, Kind::IncrementalSnapshot(_)) => Greater,
(Kind::IncrementalSnapshot(_), Kind::FullSnapshot) => Less,
(Kind::IncrementalSnapshot(base_slot_a), Kind::IncrementalSnapshot(base_slot_b)) => {
base_slot_a.cmp(base_slot_b)
}
}
}
#[cfg(test)]
mod tests {
use {super::*, solana_clock::Slot};
#[test]
fn test_cmp_snapshot_packages_by_priority() {
fn new(snapshot_kind: SnapshotKind, slot: Slot) -> SnapshotPackage {
SnapshotPackage {
snapshot_kind,
slot,
block_height: slot,
..SnapshotPackage::default_for_tests()
}
}
for (snapshot_package_a, snapshot_package_b, expected_result) in [
(
new(SnapshotKind::FullSnapshot, 11),
new(SnapshotKind::FullSnapshot, 22),
Less,
),
(
new(SnapshotKind::FullSnapshot, 22),
new(SnapshotKind::FullSnapshot, 22),
Equal,
),
(
new(SnapshotKind::FullSnapshot, 33),
new(SnapshotKind::FullSnapshot, 22),
Greater,
),
(
new(SnapshotKind::FullSnapshot, 22),
new(SnapshotKind::IncrementalSnapshot(88), 99),
Greater,
),
(
new(SnapshotKind::IncrementalSnapshot(11), 55),
new(SnapshotKind::IncrementalSnapshot(22), 55),
Less,
),
(
new(SnapshotKind::IncrementalSnapshot(22), 55),
new(SnapshotKind::IncrementalSnapshot(22), 55),
Equal,
),
(
new(SnapshotKind::IncrementalSnapshot(33), 55),
new(SnapshotKind::IncrementalSnapshot(22), 55),
Greater,
),
(
new(SnapshotKind::IncrementalSnapshot(22), 44),
new(SnapshotKind::IncrementalSnapshot(22), 55),
Less,
),
(
new(SnapshotKind::IncrementalSnapshot(22), 55),
new(SnapshotKind::IncrementalSnapshot(22), 55),
Equal,
),
(
new(SnapshotKind::IncrementalSnapshot(22), 66),
new(SnapshotKind::IncrementalSnapshot(22), 55),
Greater,
),
] {
let actual_result =
cmp_snapshot_packages_by_priority(&snapshot_package_a, &snapshot_package_b);
assert_eq!(expected_result, actual_result);
}
}
#[test]
fn test_cmp_snapshot_kinds_by_priority() {
for (snapshot_kind_a, snapshot_kind_b, expected_result) in [
(
SnapshotKind::FullSnapshot,
SnapshotKind::FullSnapshot,
Equal,
),
(
SnapshotKind::FullSnapshot,
SnapshotKind::IncrementalSnapshot(5),
Greater,
),
(
SnapshotKind::IncrementalSnapshot(5),
SnapshotKind::FullSnapshot,
Less,
),
(
SnapshotKind::IncrementalSnapshot(5),
SnapshotKind::IncrementalSnapshot(6),
Less,
),
(
SnapshotKind::IncrementalSnapshot(5),
SnapshotKind::IncrementalSnapshot(5),
Equal,
),
(
SnapshotKind::IncrementalSnapshot(5),
SnapshotKind::IncrementalSnapshot(4),
Greater,
),
] {
let actual_result = cmp_snapshot_kinds_by_priority(&snapshot_kind_a, &snapshot_kind_b);
assert_eq!(expected_result, actual_result);
}
}
}