blockworks-foundation
/
solana-with-rpc-optimizations
mirror of https://github.com/blockworks-foundation/solana-with-rpc-optimizations.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use the Entry API to remove the double lookup

This commit is contained in:
Greg Fitzgerald 2018-03-20 18:07:54 -06:00
parent c584a25ec9
commit 9a7cac1e07
1 changed files with 6 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/accountant.rs
View File

@ -13,6 +13,7 @@ use historian::{reserve_signature, Historian};
use recorder::Signal;
use std::sync::mpsc::SendError;
use std::collections::{HashMap, HashSet};
use std::collections::hash_map::Entry::Occupied;
use std::result;
use chrono::prelude::*;
@ -153,20 +154,14 @@ impl Accountant {
}
fn process_verified_sig(&mut self, from: PublicKey, tx_sig: Signature) -> Result<()> {
let actionable = if let Some(plan) = self.pending.get_mut(&tx_sig) {
plan.apply_witness(Witness::Signature(from));
if plan.is_complete() {
complete_transaction(&mut self.balances, &plan);
if let Occupied(mut e) = self.pending.entry(tx_sig) {
e.get_mut().apply_witness(Witness::Signature(from));
if e.get().is_complete() {
complete_transaction(&mut self.balances, e.get());
e.remove_entry();
}
plan.is_complete()
} else {
false
};
if actionable {
self.pending.remove(&tx_sig);
}
Ok(())
}