use brk_types::{TxWithHex, Txid};
use derive_more::Deref;
use rustc_hash::FxHashMap;
#[derive(Default, Deref)]
pub struct TxStore(FxHashMap<Txid, TxWithHex>);
impl TxStore {
pub fn contains(&self, txid: &Txid) -> bool {
self.0.contains_key(txid)
}
pub fn extend(&mut self, txs: FxHashMap<Txid, TxWithHex>) {
self.0.extend(txs);
}
pub fn retain_or_remove<K, R>(&mut self, mut keep: K, mut on_remove: R)
where
K: FnMut(&Txid) -> bool,
R: FnMut(&Txid, &TxWithHex),
{
self.0.retain(|txid, tx| {
if keep(txid) {
true
} else {
on_remove(txid, tx);
false
}
});
}
}