From 24fc2d888f429677d9c41e37e87d2b1d26b31193 Mon Sep 17 00:00:00 2001 From: Dominik Werder Date: Fri, 16 May 2025 14:53:28 +0200 Subject: [PATCH] Limit connection count --- crates/scyllaconn/src/conn.rs | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/crates/scyllaconn/src/conn.rs b/crates/scyllaconn/src/conn.rs index 01c285f..16e21ab 100644 --- a/crates/scyllaconn/src/conn.rs +++ b/crates/scyllaconn/src/conn.rs @@ -1,10 +1,11 @@ -use netpod::log::*; use netpod::ScyllaConfig; +use netpod::log::*; use scylla::client::execution_profile::ExecutionProfileBuilder; use scylla::client::session::Session; use scylla::client::session_builder::SessionBuilder; use scylla::errors::NewSessionError; use scylla::statement::Consistency; +use std::num::NonZero; use std::sync::Arc; autoerr::create_error_v1!( @@ -25,6 +26,7 @@ pub async fn create_scy_session(scyconf: &ScyllaConfig) -> Result, pub async fn create_scy_session_no_ks(scyconf: &ScyllaConfig) -> Result { info!("creating scylla connection"); let scy = SessionBuilder::new() + .pool_size(scylla::client::PoolSize::PerHost(NonZero::new(4).unwrap())) .known_nodes(&scyconf.hosts) .default_execution_profile_handle( ExecutionProfileBuilder::default()