feat(task): add an interface to query thread info (#464)

XiaoMi · May 21, 2020 · cd53ad6 · cd53ad6
1 parent a769521
commit cd53ad6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/include/dsn/tool-api/task.h b/include/dsn/tool-api/task.h
@@ -614,6 +614,8 @@ class aio_task : public task
 };
 typedef dsn::ref_ptr<aio_task> aio_task_ptr;
 
+const std::vector<task_worker *> &get_threadpool_threads_info(threadpool_code code);
+
 // ------------------------ inline implementations --------------------
 __inline /*static*/ void task::check_tls_dsn()
 {

diff --git a/src/core/core/task.cpp b/src/core/core/task.cpp
@@ -439,6 +439,11 @@ void task::enqueue(task_worker_pool *pool)
     pool->enqueue(this);
 }
 
+const std::vector<task_worker *> &get_threadpool_threads_info(threadpool_code code)
+{
+    return dsn::task::get_current_node2()->computation()->get_pool(code)->workers();
+}
+
 timer_task::timer_task(
     task_code code, const task_handler &cb, int interval_milliseconds, int hash, service_node *node)
     : task(code, hash, node), _interval_milliseconds(interval_milliseconds), _cb(cb)