fix: update code to improve exception handling

https://telecominfraproject.atlassian.net/browse/WIFI-14038 Summary of changes: - Modified code to cover zero sized queues under exception handling. Signed-off-by: Ivan Chvets <[email protected]>
Telecominfraproject · Aug 29, 2024 · 9c5aeda · 9c5aeda
1 parent 783ec99
commit 9c5aeda
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/framework/KafkaManager.cpp b/src/framework/KafkaManager.cpp
@@ -118,6 +118,10 @@ namespace OpenWifi {
 						Producer.poll((std::chrono::milliseconds) 0);
 					}
 				}
+				if (Queue_.size() == 0) {
+					// message queue is empty, flush all previously sent messages
+					Producer.flush();
+				}
 			} catch (const cppkafka::HandleException &E) {
 				poco_warning(Logger_,
 							 fmt::format("Caught a Kafka exception (producer): {}", E.what()));
@@ -126,10 +130,6 @@ namespace OpenWifi {
 			} catch (...) {
 				poco_error(Logger_, "std::exception");
 			}
-			if (Queue_.size() == 0) {
-				// message queue is empty, flush all previously sent messages
-				Producer.flush();
-			}
 			Note = Queue_.waitDequeueNotification();
 		}
 		Producer.flush();