ddsperf option for stopping when no peers remain

Signed-off-by: Erik Boasson <[email protected]>

Author: eboasson

src/ddsrt/CMakeLists.txt

@@ -27,6 +27,9 @@ set(source_dir "${CMAKE_CURRENT_SOURCE_DIR}")
 set(binary_dir "${CMAKE_CURRENT_BINARY_DIR}")
 
 add_library(ddsrt INTERFACE)
+if(ENABLE_MIMALLOC)
+  target_link_libraries(ddsrt INTERFACE mimalloc)
+endif()
 target_include_directories(
   ddsrt INTERFACE
     "$<BUILD_INTERFACE:${source_dir}/include>"
@@ -368,7 +371,7 @@ set(DDSRT_WITH_LWIP ${WITH_LWIP})
 set(DDSRT_WITH_FREERTOS ${WITH_FREERTOS})
 
 foreach(feature TCP_TLS SECURITY LIFESPAN DEADLINE_MISSED NETWORK_PARTITIONS
-                TYPELIB TYPE_DISCOVERY TOPIC_DISCOVERY QOS_PROVIDER)
+                TYPELIB TYPE_DISCOVERY TOPIC_DISCOVERY QOS_PROVIDER MIMALLOC)
   set(DDS_HAS_${feature} ${ENABLE_${feature}})
 endforeach()
 

src/tools/ddsperf/ddsperf.c

@@ -191,6 +191,9 @@ static bool sublatency = false;
 /* Use writer loans (only for memcpy-able types) */
 static bool use_writer_loan = false;
 
+/* Stop when no peers remain */
+static bool stop_when_lonely = false;
+
 /* Event queue for processing discovery events (data available on
    DCPSParticipant, subscription & publication matched)
    asynchronously to avoid deadlocking on creating a reader from
@@ -1919,6 +1922,7 @@ OPTIONS:\n\
                       from -Qmaxwait:DUR because that doesn't delay starting\n\
                       and doesn't terminate the process before doing\n\
                       anything.)\n\
+  -0                  stop when no peers remain\n\
   -1                  print \"sub\" stats every second, even when there is\n\
                       data\n\
   -X                  output extended statistics\n\
@@ -2263,11 +2267,12 @@ int main (int argc, char *argv[])
 
   argv0 = argv[0];
 
-  while ((opt = getopt (argc, argv, "1cd:D:i:n:k:ulLK:T:Q:R:Xh")) != EOF)
+  while ((opt = getopt (argc, argv, "01cd:D:i:n:k:ulLK:T:Q:R:Xh")) != EOF)
   {
     int pos;
     switch (opt)
     {
+      case '0': stop_when_lonely = true; break;
       case '1': substat_every_second = true; break;
       case 'c': collect_stats = true; break;
       case 'd': {
@@ -2729,6 +2734,11 @@ int main (int argc, char *argv[])
       {
         twakeup = pp->tdeadline;
       }
+      if (stop_when_lonely && ddsrt_avl_is_singleton (&ppants) && matchcount > 0)
+      {
+        // last peer left
+        ddsrt_atomic_st32 (&termflag, 1);
+      }
       ddsrt_mutex_unlock (&disc_lock);
     }