diff --git a/shotover-proxy/benches/windsock/cassandra/bench.rs b/shotover-proxy/benches/windsock/cassandra/bench.rs index 7ac888cec..0c53e42e9 100644 --- a/shotover-proxy/benches/windsock/cassandra/bench.rs +++ b/shotover-proxy/benches/windsock/cassandra/bench.rs @@ -11,6 +11,7 @@ use async_trait::async_trait; use aws_throwaway::Ec2Instance; use cdrs_tokio::{ cluster::{ + connection_pool::ConnectionPoolConfigBuilder, session::{ Session as CdrsTokioSession, SessionBuilder as CdrsTokioSessionBuilder, TcpSessionBuilder, @@ -742,6 +743,12 @@ impl Bench for CassandraBench { Compression::None => CdrsCompression::None, Compression::Lz4 => CdrsCompression::Lz4, }) + .with_connection_pool_config( + ConnectionPoolConfigBuilder::new() + .with_local_size(self.connection_count) + .with_remote_size(self.connection_count) + .build(), + ) .build() .await .unwrap(),