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

read-locks Turbine QUIC connection cache for lookups (#32687) 

Frequently accessed remote addresses are cached, making a shared-lock
sufficient which would reduce lock contention for the more common
cache-hit lookups.
This commit is contained in:
behzad nouri 2023-08-02 21:54:45 +00:00 committed by GitHub
parent 8c14886891
commit 349f5e0846
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
turbine/src/quic_endpoint.rs
View File

@ -294,8 +294,7 @@ async fn get_connection(
sender: Sender<(Pubkey, SocketAddr, Bytes)>,
cache: Arc<RwLock<ConnectionCache>>,
) -> Result<Connection, Error> {
let key = (remote_address, /*remote_pubkey:*/ None);
let entry = cache.write().await.entry(key).or_default().clone();
let entry = get_cache_entry(remote_address, &cache).await;
{
let connection: Option<Connection> = entry.read().await.clone();
if let Some(connection) = connection {
@ -342,6 +341,17 @@ fn get_remote_pubkey(connection: &Connection) -> Result<Pubkey, Error> {
}
}
async fn get_cache_entry(
remote_address: SocketAddr,
cache: &RwLock<ConnectionCache>,
) -> Arc<RwLock<Option<Connection>>> {
let key = (remote_address, /*remote_pubkey:*/ None);
if let Some(entry) = cache.read().await.get(&key) {
return entry.clone();
}
cache.write().await.entry(key).or_default().clone()
}
async fn cache_connection(
remote_address: SocketAddr,
remote_pubkey: Pubkey,