From a423e602aa7efbea997048a162349f16a27119b9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Florian=20L=C3=BCke?= <f.lueke@mesytec.com>
Date: Tue, 7 May 2024 14:56:25 +0200
Subject: [PATCH] push pull: can now pass ProducerCount on the command line

---
 src/mesy_nng_push_pull_main.cc | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/mesy_nng_push_pull_main.cc b/src/mesy_nng_push_pull_main.cc
index 925a1f8..24da0f1 100644
--- a/src/mesy_nng_push_pull_main.cc
+++ b/src/mesy_nng_push_pull_main.cc
@@ -144,6 +144,11 @@ int main(int argc, char *argv[])
 {
     spdlog::set_level(spdlog::level::info);
 
+    size_t ProducerCount = 10;
+
+    if (argc > 1)
+        ProducerCount = std::stoull(argv[1]);
+
     nng_socket consumerSocket = NNG_SOCKET_INITIALIZER;
 
     if (int res = nng_pull0_open(&consumerSocket))
@@ -162,10 +167,13 @@ int main(int argc, char *argv[])
 
     std::vector<std::thread> threads;
 
-    const size_t ProducerCount = 10;
+    spdlog::info("Starting {} producers.", ProducerCount);
+
     for (size_t i=0; i<ProducerCount; ++i)
         threads.emplace_back(std::thread(push_producer, producerSocket, i));
 
+    spdlog::info("Starting consumer.");
+
     // Deliberately start the consumer after the producers. There should be no buffer loss.
     threads.emplace_back(std::thread(pull_consumer, consumerSocket));