[BUG]Added support for data balancing when SSL is enabled. snapshot m…

…anager add ssl config (#5920)

* snapshot manager add ssl config

* snapshot manager add ssl config

* fix cpplint code style

* fix code style

* fix code style

---------

Co-authored-by: l00416797 <liuxudong6@huawei.com>
Co-authored-by: Wey Gu <weyl.gu@gmail.com>

src/kvstore/raftex/SnapshotManager.cpp

@@ -24,6 +24,8 @@ SnapshotManager::SnapshotManager() {
   ioThreadPool_.reset(new folly::IOThreadPoolExecutor(
       FLAGS_snapshot_io_threads,
       std::make_shared<folly::NamedThreadFactory>("snapshot-ioexecutor")));
+  connManager_.reset(
+      new thrift::ThriftClientManager<raftex::cpp2::RaftexServiceAsyncClient>(FLAGS_enable_ssl));
 }
 
 folly::Future<StatusOr<std::pair<LogID, TermID>>> SnapshotManager::sendSnapshot(
@@ -131,7 +133,7 @@ folly::Future<raftex::cpp2::SendSnapshotResponse> SnapshotManager::send(
   req.done_ref() = finished;
   auto* evb = ioThreadPool_->getEventBase();
   return folly::via(evb, [this, addr, evb, req = std::move(req)]() mutable {
-    auto client = connManager_.client(addr, evb, false, FLAGS_snapshot_send_timeout_ms);
+    auto client = connManager_->client(addr, evb, false, FLAGS_snapshot_send_timeout_ms);
     return client->future_sendSnapshot(req);
   });
 }

src/kvstore/raftex/SnapshotManager.h

@@ -12,6 +12,7 @@
 
 #include "common/base/Base.h"
 #include "common/base/StatusOr.h"
+#include "common/ssl/SSLConfig.h"
 #include "common/thrift/ThriftClientManager.h"
 #include "interface/gen-cpp2/RaftexServiceAsyncClient.h"
 #include "interface/gen-cpp2/raftex_types.h"
@@ -93,7 +94,7 @@ class SnapshotManager {
  private:
   std::unique_ptr<folly::IOThreadPoolExecutor> executor_;
   std::unique_ptr<folly::IOThreadPoolExecutor> ioThreadPool_;
-  thrift::ThriftClientManager<raftex::cpp2::RaftexServiceAsyncClient> connManager_;
+  std::unique_ptr<thrift::ThriftClientManager<raftex::cpp2::RaftexServiceAsyncClient>> connManager_;
 };
 
 }  // namespace raftex