gperftools profiling

sflow · May 31, 2023 · 65e5174 · 65e5174
1 parent 032af6a
commit 65e5174
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/src/Linux/evbus.c b/src/Linux/evbus.c
@@ -9,6 +9,10 @@ extern "C" {
 #include "util.h"
 #include "evbus.h"
 
+#ifdef GPROF
+#include "gperftools/profiler.h"
+#endif
+
   // only one running bus in each thread - keep track with thread-local var
   // so we can always know what the current "home" bus is and detect
   // inter-bus (inter-thread) messages automatically in EVEventTx
@@ -463,6 +467,10 @@ extern "C" {
   }
 
   static void *busRun(void *magic) {
+#ifdef GPROF
+    myDebug(1, "GPROF ProfilerRegisterThread()");
+    ProfilerRegisterThread();
+#endif
     EVBus *bus = (EVBus *)magic;
     EVMod *mod = bus->root->rootModule;
     assert(bus->running == NO);

diff --git a/src/Linux/hsflowd.c b/src/Linux/hsflowd.c
@@ -1832,11 +1832,6 @@ extern "C" {
 #ifdef UTHEAP
     UTHeapInit();
 #endif
-
-#ifdef GPROF
-    myDebug(1, "GPROF ProfilerStart()");
-    ProfilerStart(NULL);
-#endif
 
     // open syslog
     openlog(HSP_DAEMON_NAME, LOG_CONS, LOG_USER);
@@ -1882,6 +1877,11 @@ extern "C" {
       exit(EXIT_FAILURE);
     }
 
+#ifdef GPROF
+    myDebug(1, "GPROF ProfilerStart()");
+    ProfilerStart(NULL);
+#endif
+
     if(getDaemon()) {
       // fork to daemonize
       pid_t pid = fork();
@@ -2288,6 +2288,7 @@ extern "C" {
 
 #ifdef GPROF
     myDebug(1, "GPROF ProfilerStop()");
+    ProfilerFlush();
     ProfilerStop();
 #endif