Skip to content

Commit 76ddd16

Browse files
committed
Get the pool from the pool map given the server instance
1 parent 0861a1e commit 76ddd16

File tree

2 files changed

+5
-6
lines changed

2 files changed

+5
-6
lines changed

lib/src/routing/connection_registry.rs

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -258,11 +258,10 @@ pub(crate) fn start_background_updater(
258258

259259
impl ConnectionRegistry {
260260
/// Retrieve the pool for a specific server and database.
261-
pub fn get_pool(&self, server: &BoltServer, db: Option<Database>) -> Option<ConnectionPool> {
262-
let pair = self.servers(db.clone());
263-
pair.iter()
264-
.find(|bs| *bs == server)
265-
.and_then(|bs| self.pool_registry.get(bs).map(|pool| pool.value().clone()))
261+
pub fn get_pool(&self, server: &BoltServer) -> Option<ConnectionPool> {
262+
self.pool_registry
263+
.get(server)
264+
.map(|pool| pool.value().clone())
266265
}
267266

268267
/// Mark a server as available for a specific database.

lib/src/routing/routed_connection_manager.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ impl RoutedConnectionManager {
9999
_ => self.select_reader(db.clone()),
100100
} {
101101
debug!("requesting connection for server: {:?}", server);
102-
if let Some(pool) = self.connection_registry.get_pool(&server, db.clone()) {
102+
if let Some(pool) = self.connection_registry.get_pool(&server) {
103103
match pool.get().await {
104104
Ok(connection) => return Ok(connection),
105105
Err(e) => {

0 commit comments

Comments
 (0)