mirrors/electrs
1
0
Fork 0
mirror of https://github.com/romanz/electrs.git synced 2024-11-19 09:54:09 +01:00
Code Issues Projects Releases Wiki Activity

Scan mempool transaction during status query

Browse Source
This commit is contained in:
Roman Zeyde 2018-05-21 11:34:36 +03:00
parent 5941edbbc6
commit de884ac06b
No known key found for this signature in database
GPG Key ID: 87CAE5FA46917CBB
1 changed files with 36 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
src/query.rs
View File

@ -146,9 +146,9 @@ impl<'a> Query<'a> {
txns txns
} }
fn find_spending_input(&self, funding: &FundingOutput) -> Option<SpendingInput> { fn find_spending_input(&self, store: &Store, funding: &FundingOutput) -> Option<SpendingInput> {
let spending_txns: Vec<TxnHeight> = self.load_txns(txids_by_funding_output( let spending_txns: Vec<TxnHeight> = self.load_txns(txids_by_funding_output(
self.store, store,
&funding.txn_id, &funding.txn_id,
funding.output_index, funding.output_index,
)); ));
@ -190,20 +190,19 @@ impl<'a> Query<'a> {
result result
} }
pub fn status(&self, script_hash: &[u8]) -> Status { fn confirmed_status(&self, script_hash: &[u8]) -> Status {
let mut status = Status { let mut status = Status {
balance: 0, balance: 0,
funding: vec![], funding: vec![],
spending: vec![], spending: vec![],
}; };
for t in self.load_txns(txids_by_script_hash(self.store, script_hash)) { for t in self.load_txns(txids_by_script_hash(self.store, script_hash)) {
status status
.funding .funding
.extend(self.find_funding_outputs(&t, script_hash)); .extend(self.find_funding_outputs(&t, script_hash));
} }
for funding_output in &status.funding { for funding_output in &status.funding {
if let Some(spent) = self.find_spending_input(&funding_output) { if let Some(spent) = self.find_spending_input(self.store, &funding_output) {
status.spending.push(spent); status.spending.push(spent);
} else { } else {
status.balance += funding_output.value; status.balance += funding_output.value;
@ -212,6 +211,38 @@ impl<'a> Query<'a> {
status status
} }
fn mempool_status(&self, script_hash: &[u8], confirmed_status: &Status) -> Status {
let mut status = Status {
balance: 0,
funding: vec![],
spending: vec![],
};
let mempool_store = self.tracker.read().unwrap().build_index();
for t in self.load_txns(txids_by_script_hash(&*mempool_store, script_hash)) {
status
.funding
.extend(self.find_funding_outputs(&t, script_hash));
}
// TODO: dedup outputs (somehow) both confirmed and in mempool (e.g. reorg?)
for funding_output in status.funding.iter().chain(confirmed_status.funding.iter()) {
if let Some(spent) = self.find_spending_input(&*mempool_store, &funding_output) {
status.spending.push(spent);
} else {
status.balance += funding_output.value;
}
}
status
}
pub fn status(&self, script_hash: &[u8]) -> Status {
let mut status = self.confirmed_status(script_hash);
let mempool_status = self.mempool_status(script_hash, &status);
status.balance += mempool_status.balance;
status.funding.extend(mempool_status.funding);
status.spending.extend(mempool_status.spending);
status
}
pub fn get_tx(&self, tx_hash: &Sha256dHash) -> Transaction { pub fn get_tx(&self, tx_hash: &Sha256dHash) -> Transaction {
self.daemon self.daemon
.gettransaction(tx_hash) .gettransaction(tx_hash)