110 lines
3.5 KiB
Rust
110 lines
3.5 KiB
Rust
use solana_client::rpc_response::{Response, RpcKeyedAccount};
|
|
use solana_sdk::{account::AccountSharedData, pubkey::Pubkey};
|
|
|
|
use std::time::Instant;
|
|
use std::{str::FromStr, sync::Arc};
|
|
use tracing::*;
|
|
|
|
use crate::chain_data;
|
|
|
|
#[derive(Clone)]
|
|
pub struct AccountUpdate {
|
|
pub pubkey: Pubkey,
|
|
pub slot: u64,
|
|
pub account: AccountSharedData,
|
|
pub reception_time: Instant,
|
|
}
|
|
|
|
impl AccountUpdate {
|
|
pub fn from_rpc(rpc: Response<RpcKeyedAccount>) -> anyhow::Result<Self> {
|
|
let pubkey = Pubkey::from_str(&rpc.value.pubkey)?;
|
|
let account = rpc
|
|
.value
|
|
.account
|
|
.decode()
|
|
.ok_or_else(|| anyhow::anyhow!("could not decode account"))?;
|
|
Ok(AccountUpdate {
|
|
pubkey,
|
|
slot: rpc.context.slot,
|
|
account,
|
|
reception_time: Instant::now(),
|
|
})
|
|
}
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct ChainSlotUpdate {
|
|
pub slot_update: Arc<solana_client::rpc_response::SlotUpdate>,
|
|
pub reception_time: Instant,
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub enum Message {
|
|
Account(AccountUpdate),
|
|
Snapshot(Vec<AccountUpdate>),
|
|
Slot(ChainSlotUpdate),
|
|
}
|
|
|
|
impl Message {
|
|
pub fn update_chain_data(&self, chain: &mut chain_data::ChainData) {
|
|
use chain_data::*;
|
|
match self {
|
|
Message::Account(account_write) => {
|
|
trace!("websocket account message");
|
|
chain.update_account(
|
|
account_write.pubkey,
|
|
AccountData {
|
|
slot: account_write.slot,
|
|
account: account_write.account.clone(),
|
|
write_version: 1,
|
|
},
|
|
);
|
|
}
|
|
Message::Snapshot(snapshot) => {
|
|
for account_update in snapshot {
|
|
chain.update_account(
|
|
account_update.pubkey,
|
|
chain_data::AccountData {
|
|
slot: account_update.slot,
|
|
account: account_update.account.clone(),
|
|
write_version: 0,
|
|
},
|
|
);
|
|
}
|
|
}
|
|
Message::Slot(slot_update) => {
|
|
trace!("websocket slot message");
|
|
let slot_update = match *(slot_update.slot_update) {
|
|
solana_client::rpc_response::SlotUpdate::CreatedBank {
|
|
slot, parent, ..
|
|
} => Some(SlotData {
|
|
slot,
|
|
parent: Some(parent),
|
|
status: SlotStatus::Processed,
|
|
chain: 0,
|
|
}),
|
|
solana_client::rpc_response::SlotUpdate::OptimisticConfirmation {
|
|
slot,
|
|
..
|
|
} => Some(SlotData {
|
|
slot,
|
|
parent: None,
|
|
status: SlotStatus::Confirmed,
|
|
chain: 0,
|
|
}),
|
|
solana_client::rpc_response::SlotUpdate::Root { slot, .. } => Some(SlotData {
|
|
slot,
|
|
parent: None,
|
|
status: SlotStatus::Rooted,
|
|
chain: 0,
|
|
}),
|
|
_ => None,
|
|
};
|
|
if let Some(update) = slot_update {
|
|
chain.update_slot(update);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|