chore: change num connections config to usize

momentohq · Oct 9, 2024 · 9612858 · 9612858
1 parent 036c129
commit 9612858
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/sdk/src/cache/cache_client_builder.rs b/sdk/src/cache/cache_client_builder.rs
@@ -62,7 +62,7 @@ impl CacheClientBuilder<NeedsCredentialProvider> {
 }
 
 impl CacheClientBuilder<ReadyToBuild> {
-    pub fn with_num_connections(self, num_connections: u32) -> CacheClientBuilder<ReadyToBuild> {
+    pub fn with_num_connections(self, num_connections: usize) -> CacheClientBuilder<ReadyToBuild> {
         let grpc_configuration = self.0.configuration.transport_strategy.grpc_configuration;
         let transport_strategy = TransportStrategy {
             grpc_configuration: GrpcConfiguration {

diff --git a/sdk/src/config/grpc_configuration.rs b/sdk/src/config/grpc_configuration.rs
@@ -7,7 +7,7 @@ pub struct GrpcConfiguration {
     /// with a DeadlineExceeded error.
     pub(crate) deadline: Duration,
     /// The number of grpc channels (TCP connections) to create
-    pub(crate) num_channels: u32,
+    pub(crate) num_channels: usize,
     /// Indicates whether the client should send keep-alive pings.
     ///
     /// NOTE: keep-alives are very important for long-lived server environments where there may be
@@ -40,7 +40,7 @@ pub struct NeedsDeadline(());
 /// The state of the GrpcConfigurationBuilder when it is ready to build a GrpcConfiguration.
 pub struct ReadyToBuild {
     deadline: Duration,
-    num_channels: u32,
+    num_channels: usize,
     keep_alive_while_idle: Option<bool>,
     keep_alive_interval: Option<Duration>,
     keep_alive_timeout: Option<Duration>,
@@ -61,7 +61,7 @@ impl GrpcConfigurationBuilder<NeedsDeadline> {
 }
 
 impl GrpcConfigurationBuilder<ReadyToBuild> {
-    pub(crate) fn num_channels(mut self, num_channels: u32) -> Self {
+    pub(crate) fn num_channels(mut self, num_channels: usize) -> Self {
         self.0.num_channels = num_channels;
         self
     }