Fix compilation issue after rebase with main

Author: madchicken

lib/src/query.rs

@@ -117,6 +117,7 @@ impl Query {
             query,
             operation,
             fetch_size,
+            db,
         }
     }
 
@@ -233,6 +234,7 @@ pub(crate) struct RetryableQuery<'a> {
     query: Query,
     operation: Operation,
     fetch_size: Option<usize>,
+    db: Option<Database>,
 }
 
 impl<'a> RetryableQuery<'a> {
@@ -270,7 +272,10 @@ impl<'a> RetryableQuery<'a> {
 
     async fn connect(&self) -> QueryResult<ManagedConnection> {
         // an error when retrieving a connection is considered permanent
-        self.pool.get(Some(self.operation)).await.map_err(Retry::No)
+        self.pool
+            .get(Some(self.operation), self.db.clone())
+            .await
+            .map_err(Retry::No)
     }
 }
 

lib/src/routing/connection_registry.rs

@@ -557,9 +557,10 @@ mod tests {
 
         let _ = registry.servers(Some("db1".into())); // ensure db1 is initialized
         let _ = registry.servers(Some("db2".into())); // ensure db2 is initialized
-        let ttl = refresh_all_routing_tables(config.clone(), registry.clone(), provider.clone(), &[])
-            .await
-            .unwrap();
+        let ttl =
+            refresh_all_routing_tables(config.clone(), registry.clone(), provider.clone(), &[])
+                .await
+                .unwrap();
 
         let servers = registry.servers(Some("db1".into()));
         assert_eq!(ttl, 200); // must be the min of both