From aa6163937d47ca2e18259e311cca5e824477ed29 Mon Sep 17 00:00:00 2001
From: wxyu <xy.wang@zilliz.com>
Date: Fri, 23 Aug 2019 11:35:36 +0800
Subject: [PATCH] MS-405 Add delete task support

Former-commit-id: a1982d73b6d2f2bfdaa6e134f5efc401b5e4a113
---
 cpp/src/scheduler/ResourceMgr.cpp | 11 +++++++++++
 cpp/src/scheduler/ResourceMgr.h   |  6 ++++++
 2 files changed, 17 insertions(+)

diff --git a/cpp/src/scheduler/ResourceMgr.cpp b/cpp/src/scheduler/ResourceMgr.cpp
index cf54472732..f341821392 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 c3f900b2db..54ff551b6b 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;