fix: Added retry channel (#33)

aneoconsulting · Oct 11, 2023 · 69edcdf · 69edcdf
2 parents e187295 + 554c674
commit 69edcdf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/ArmoniK.SDK.Client/src/ChannelPool.cpp b/ArmoniK.SDK.Client/src/ChannelPool.cpp
@@ -1,6 +1,7 @@
 #include "ChannelPool.h"
 
 #include <armonik/common/options/ControlPlane.h>
+#include <armonik/common/utils/ChannelArguments.h>
 #include <grpcpp/create_channel.h>
 #include <utility>
 
@@ -27,14 +28,16 @@ std::shared_ptr<grpc::Channel> ChannelPool::AcquireChannel() {
       return channel;
     }
   }
-
   std::string endpoint(properties_.configuration.get_control_plane().getEndpoint());
   auto scheme_delim = endpoint.find("://");
   if (scheme_delim != std::string::npos) {
     endpoint = endpoint.substr(scheme_delim + 3);
   }
   // TODO Handle TLS/mTLS
-  channel = grpc::CreateChannel(endpoint, grpc::InsecureChannelCredentials());
+  channel = grpc::CreateCustomChannel(
+      endpoint, grpc::InsecureChannelCredentials(),
+      armonik::api::common::utils::getChannelArguments(
+          static_cast<armonik::api::common::utils::Configuration>(properties_.configuration)));
   logger_.log(armonik::api::common::logger::Level::Debug, "Created and acquired new channel from pool");
   return channel;
 }

diff --git a/tools/common.sh b/tools/common.sh
@@ -1,5 +1,5 @@
 #!/usr/bin/env bash
 
 # shellcheck disable=SC2034
-ARMONIK_API_VERSION_DEFAULT="3.13.1"
+ARMONIK_API_VERSION_DEFAULT="3.13.2"
 ARMONIK_SDK_VERSION_DEFAULT="0.1.0-local"