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

Simplify code and use match to harden logic (#33409) 

addressed more feedback from Jon: Simplify code and use match to harden logic in connection cache
This commit is contained in:
Lijun Wang 2023-09-27 09:35:19 -07:00 committed by GitHub
parent 3608378097
commit 3fbfa0e0da
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 28 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
connection-cache/src/connection_cache.rs
View File

@ -196,8 +196,8 @@ where
}
fn create_connection_internal(
config: &Arc<C>,
connection_manager: &Arc<M>,
config: &C,
connection_manager: &M,
map: &mut std::sync::RwLockWriteGuard<'_, IndexMap<SocketAddr, P>>,
addr: &SocketAddr,
connection_pool_size: usize,
@ -276,11 +276,13 @@ where
} = match map.get(addr) {
Some(pool) => {
let pool_status = pool.check_pool_status(self.connection_pool_size);
if matches!(pool_status, PoolStatus::Empty) {
match pool_status {
PoolStatus::Empty => {
// create more connection and put it in the pool
drop(map);
self.create_connection(&mut lock_timing_ms, addr)
} else {
}
PoolStatus::PartiallyFull | PoolStatus::Full => {
let connection = pool.borrow_connection();
if matches!(pool_status, PoolStatus::PartiallyFull) {
debug!("Creating connection async for {addr}");
@ -304,6 +306,7 @@ where
}
}
}
}
None => {
// Upgrade to write access by dropping read lock and acquire write lock
drop(map);