diff --git a/cpp/src/scheduler/ResourceMgr.cpp b/cpp/src/scheduler/ResourceMgr.cpp index cf54472732606e5381bc817f1cb5ff27927104a5..f341821392e746c729289ae93a91b1f5b15278ff 100644 --- a/cpp/src/scheduler/ResourceMgr.cpp +++ b/cpp/src/scheduler/ResourceMgr.cpp @@ -17,6 +17,17 @@ ResourceMgr::ResourceMgr() } +uint64_t +ResourceMgr::GetNumOfComputeResource() { + uint64_t count = 0; + for (auto &res : resources_) { + if (res->HasExecutor()) { + ++count; + } + } + return count; +} + ResourceWPtr ResourceMgr::Add(ResourcePtr &&resource) { ResourceWPtr ret(resource); diff --git a/cpp/src/scheduler/ResourceMgr.h b/cpp/src/scheduler/ResourceMgr.h index c3f900b2db25792c17ff61be20d6cef04933ef17..54ff551b6b36fdf41bfe76a54b38fa41672d7d95 100644 --- a/cpp/src/scheduler/ResourceMgr.h +++ b/cpp/src/scheduler/ResourceMgr.h @@ -35,6 +35,12 @@ public: return disk_resources_; } + /* + * Return account of resource which enable executor; + */ + uint64_t + GetNumOfComputeResource(); + /* * Add resource into Resource Management; * Generate functions on events;