Set format if one is not set

cms-sw · Feb 28, 2023 · 7a3b308 · 7a3b308
1 parent 9d3e42b
commit 7a3b308
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/PerfTools/JeProf/src/jeprof.cc b/PerfTools/JeProf/src/jeprof.cc
@@ -26,11 +26,25 @@ namespace jeprof {
     return enable_prof;
   }
 
+  std::string replace(const std::string &s, const char *pat, std::string_view val) {
+    size_t pos = 0;
+    size_t patlen = strlen(pat);
+    std::string result = s;
+    while ((pos = result.find(pat, pos)) != std::string::npos) {
+      result.replace(pos, patlen, val.data());
+      pos = pos - patlen + val.size();
+    }
+
+    return result;
+  }
 
   void makeDump(const std::string &format, std::string_view moduleLabel) {
-    if (!have_jemalloc_and_prof || format.empty())
+    if (!have_jemalloc_and_prof)
       return;
-    std::string final("heap_info.out");
+    std::string final(format);
+    if (format.empty())
+       final = "%M.heap_info.out";
+    final = replace(final, "%M", moduleLabel);
     const char *fileName = final.c_str();
     mallctl("prof.dump", nullptr, nullptr, &fileName, sizeof(const char *));
   }