[Pulsar C++ Client] make pool connections configurable in perf tools (a…

…pache#8913) ### Motivation make pool connections configurable in c++ performance test.
abhiQmar · Feb 15, 2021 · 530c128 · 530c128
1 parent fd9875c
commit 530c128
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/pulsar-client-cpp/perf/PerfConsumer.cc b/pulsar-client-cpp/perf/PerfConsumer.cc
@@ -68,6 +68,7 @@ struct Arguments {
     int receiverQueueSize;
     int ioThreads;
     int listenerThreads;
+    bool poolConnections;
     std::string encKeyName;
     std::string encKeyValueFile;
 };
@@ -174,7 +175,7 @@ void startPerfConsumer(const Arguments& args) {
         conf.setAuth(auth);
     }
 
-    Client client(pulsar::PulsarFriend::getClient(args.serviceURL, conf, false));
+    Client client(pulsar::PulsarFriend::getClient(args.serviceURL, conf, args.poolConnections));
 
     ConsumerConfiguration consumerConf;
     consumerConf.setMessageListener(messageListener);
@@ -299,6 +300,9 @@ int main(int argc, char** argv) {
         ("listener-threads,l", po::value<int>(&args.listenerThreads)->default_value(1),
          "Number of listener threads")  //
 
+        ("pool-connections", po::value<bool>(&args.poolConnections)->default_value(false),
+         "whether pool connections used")  //
+
         ("encryption-key-name,k", po::value<std::string>(&args.encKeyName)->default_value(""),
          "The private key name to decrypt payload")  //
 

diff --git a/pulsar-client-cpp/perf/PerfProducer.cc b/pulsar-client-cpp/perf/PerfProducer.cc
@@ -62,6 +62,7 @@ struct Arguments {
     unsigned int batchingMaxMessages;
     long batchingMaxAllowedSizeInBytes;
     long batchingMaxPublishDelayMs;
+    bool poolConnections;
     std::string encKeyName;
     std::string encKeyValueFile;
     std::string compression;
@@ -278,6 +279,9 @@ int main(int argc, char** argv) {
          po::value<long>(&args.batchingMaxPublishDelayMs)->default_value(3000),
          "Use only is batch-size > 1, Default is 3 seconds")  //
 
+        ("pool-connections", po::value<bool>(&args.poolConnections)->default_value(false),
+         "whether pool connections used")  //
+
         ("encryption-key-name,k", po::value<std::string>(&args.encKeyName)->default_value(""),
          "The public key name to encrypt payload")  //
 
@@ -376,7 +380,7 @@ int main(int argc, char** argv) {
         conf.setAuth(auth);
     }
 
-    pulsar::Client client(pulsar::PulsarFriend::getClient(args.serviceURL, conf, false));
+    pulsar::Client client(pulsar::PulsarFriend::getClient(args.serviceURL, conf, args.poolConnections));
 
     std::atomic<bool> exitCondition(false);
     startPerfProducer(args, producerConf, client, exitCondition);