Changed the managment connection to only be preferred but to fallback…

… to the user connection if needed
amazon-contributing · Dec 21, 2023 · fe1237c · fe1237c
1 parent 66fd410
commit fe1237c
Show file tree

Hide file tree

Showing 2 changed files with 271 additions and 76 deletions.
diff --git a/redis/src/cluster_async/connections_container.rs b/redis/src/cluster_async/connections_container.rs
@@ -37,6 +37,7 @@ where
             ConnectionType::User => self.user_connection.clone(),
             ConnectionType::PreferManagement => self
                 .management_connection
+                .clone()
                 .unwrap_or_else(|| self.user_connection.clone()),
         }
     }
@@ -631,7 +632,7 @@ mod tests {
     fn get_random_management_connections() {
         let container = create_container_with_strategy(ReadFromReplicaStrategy::RoundRobin);
         let mut random_connections: Vec<_> = container
-            .random_connections(1000, ConnectionType::Management)
+            .random_connections(1000, ConnectionType::PreferManagement)
             .map(|pair| pair.1)
             .collect();
         random_connections.sort();