From e9f08cc797420cd81c70c80046f2b74942fb2dd7 Mon Sep 17 00:00:00 2001
From: Lucas Kent <rubickent@gmail.com>
Date: Tue, 6 Feb 2024 18:48:58 +1100
Subject: [PATCH] windsock: fix cdrs connection count

---
 shotover-proxy/benches/windsock/cassandra/bench.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/shotover-proxy/benches/windsock/cassandra/bench.rs b/shotover-proxy/benches/windsock/cassandra/bench.rs
index f9c980716..099381e7b 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,
@@ -735,6 +736,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(),