windsock: fix cdrs connection count (#1456)

shotover · Feb 7, 2024 · 05fd8bc · 05fd8bc
1 parent 15e030c
commit 05fd8bc
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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(),