Add helper to rename all threads of a ctpl::thread_pool

Author: codablock

src/util/threadnames.cpp

@@ -15,6 +15,10 @@
 
 #include <util/threadnames.h>
 
+#include "ctpl.h"
+#include "utiltime.h"
+#include "tinyformat.h"
+
 #ifdef HAVE_SYS_PRCTL_H
 #include <sys/prctl.h> // For prctl, PR_SET_NAME, PR_GET_NAME
 #endif
@@ -64,3 +68,22 @@ void util::ThreadSetInternalName(std::string&& name)
 {
     SetInternalName(std::move(name));
 }
+
+void RenameThreadPool(ctpl::thread_pool& tp, const char* baseName)
+{
+    auto cond = std::make_shared<std::condition_variable>();
+    auto mutex = std::make_shared<std::mutex>();
+    std::atomic<int> doneCnt(0);
+    for (int i = 0; i < tp.size(); i++) {
+        tp.push([baseName, i, cond, mutex, &doneCnt](int threadId) {
+            util::ThreadRename(strprintf("%s-%d", baseName, i).c_str());
+            doneCnt++;
+            std::unique_lock<std::mutex> l(*mutex);
+            cond->wait(l);
+        });
+    }
+    while (doneCnt != tp.size()) {
+        MilliSleep(10);
+    }
+    cond->notify_all();
+}

src/util/threadnames.h

@@ -23,4 +23,9 @@ const std::string& ThreadGetInternalName();
 
 } // namespace util
 
+namespace ctpl {
+    class thread_pool;
+}
+void RenameThreadPool(ctpl::thread_pool& tp, const char* baseName);
+
 #endif // BITCOIN_UTIL_THREADNAMES_H