89c45a57f8
Problem Slot status can be used of in other scenarios in addition to account information such as transactions, blocks. The current implementation is too tightly coupled. Summary of Changes Decouple the slot status notification from accounts notification. Created a new slot status notification module.
75 lines
2.0 KiB
Rust
75 lines
2.0 KiB
Rust
use {log::*, std::collections::HashSet};
|
|
|
|
#[derive(Debug)]
|
|
pub(crate) struct AccountsSelector {
|
|
pub accounts: HashSet<Vec<u8>>,
|
|
pub owners: HashSet<Vec<u8>>,
|
|
pub select_all_accounts: bool,
|
|
}
|
|
|
|
impl AccountsSelector {
|
|
pub fn default() -> Self {
|
|
AccountsSelector {
|
|
accounts: HashSet::default(),
|
|
owners: HashSet::default(),
|
|
select_all_accounts: true,
|
|
}
|
|
}
|
|
|
|
pub fn new(accounts: &[String], owners: &[String]) -> Self {
|
|
info!(
|
|
"Creating AccountsSelector from accounts: {:?}, owners: {:?}",
|
|
accounts, owners
|
|
);
|
|
|
|
let select_all_accounts = accounts.iter().any(|key| key == "*");
|
|
if select_all_accounts {
|
|
return AccountsSelector {
|
|
accounts: HashSet::default(),
|
|
owners: HashSet::default(),
|
|
select_all_accounts,
|
|
};
|
|
}
|
|
let accounts = accounts
|
|
.iter()
|
|
.map(|key| bs58::decode(key).into_vec().unwrap())
|
|
.collect();
|
|
let owners = owners
|
|
.iter()
|
|
.map(|key| bs58::decode(key).into_vec().unwrap())
|
|
.collect();
|
|
AccountsSelector {
|
|
accounts,
|
|
owners,
|
|
select_all_accounts,
|
|
}
|
|
}
|
|
|
|
pub fn is_account_selected(&self, account: &[u8], owner: &[u8]) -> bool {
|
|
self.select_all_accounts || self.accounts.contains(account) || self.owners.contains(owner)
|
|
}
|
|
|
|
/// Check if any account is of interested at all
|
|
pub fn is_enabled(&self) -> bool {
|
|
self.select_all_accounts || !self.accounts.is_empty() || !self.owners.is_empty()
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
pub(crate) mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_create_accounts_selector() {
|
|
AccountsSelector::new(
|
|
&["9xQeWvG816bUx9EPjHmaT23yvVM2ZWbrrpZb9PusVFin".to_string()],
|
|
&[],
|
|
);
|
|
|
|
AccountsSelector::new(
|
|
&[],
|
|
&["9xQeWvG816bUx9EPjHmaT23yvVM2ZWbrrpZb9PusVFin".to_string()],
|
|
);
|
|
}
|
|
}
|