rpc-sts: dedupe before initial send (#22856)

(cherry picked from commit 9f1f7aff2b)

Co-authored-by: Trent Nelson <trent@solana.com>
This commit is contained in:
mergify[bot]
2022-02-01 01:54:54 +00:00
committed by GitHub
parent 6c47a98945
commit f83cb74509

View File

@ -5,7 +5,7 @@ use {
solana_runtime::{bank::Bank, bank_forks::BankForks}, solana_runtime::{bank::Bank, bank_forks::BankForks},
solana_sdk::{hash::Hash, nonce_account, pubkey::Pubkey, signature::Signature}, solana_sdk::{hash::Hash, nonce_account, pubkey::Pubkey, signature::Signature},
std::{ std::{
collections::HashMap, collections::hash_map::{Entry, HashMap},
net::{SocketAddr, UdpSocket}, net::{SocketAddr, UdpSocket},
sync::{ sync::{
mpsc::{Receiver, RecvTimeoutError}, mpsc::{Receiver, RecvTimeoutError},
@ -144,6 +144,9 @@ impl SendTransactionService {
Err(RecvTimeoutError::Timeout) => {} Err(RecvTimeoutError::Timeout) => {}
Ok(transaction_info) => { Ok(transaction_info) => {
inc_new_counter_info!("send_transaction_service-recv-tx", 1); inc_new_counter_info!("send_transaction_service-recv-tx", 1);
let transactions_len = transactions.len();
let entry = transactions.entry(transaction_info.signature);
if let Entry::Vacant(_) = entry {
let addresses = leader_info.as_ref().map(|leader_info| { let addresses = leader_info.as_ref().map(|leader_info| {
leader_info.get_leader_tpus(config.leader_forward_count) leader_info.get_leader_tpus(config.leader_forward_count)
}); });
@ -163,12 +166,15 @@ impl SendTransactionService {
&transaction_info.wire_transaction, &transaction_info.wire_transaction,
); );
} }
if transactions.len() < MAX_TRANSACTION_QUEUE_SIZE { if transactions_len < MAX_TRANSACTION_QUEUE_SIZE {
inc_new_counter_info!("send_transaction_service-insert-tx", 1); inc_new_counter_info!("send_transaction_service-insert-tx", 1);
transactions.insert(transaction_info.signature, transaction_info); entry.or_insert(transaction_info);
} else { } else {
datapoint_warn!("send_transaction_service-queue-overflow"); datapoint_warn!("send_transaction_service-queue-overflow");
} }
} else {
inc_new_counter_info!("send_transaction_service-recv-duplicate", 1);
}
} }
} }