MS-611 Add resources validity check in ResourceMgr

Former-commit-id: 534264d0ab113ac14f53d53b96060bda5195e022

MS-611 Add resources validity check in ResourceMgr
Former-commit-id: 534264d0ab113ac14f53d53b96060bda5195e022
02a76910 · wxyu · 0bb9b799 · 02a76910 · 02a76910
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

cpp/src/scheduler/ResourceMgr.cpp cpp/src/scheduler/ResourceMgr.cpp +2 -0

cpp/unittest/db/utils.cpp cpp/unittest/db/utils.cpp +1 -0

未找到文件。
--- a/cpp/src/scheduler/ResourceMgr.cpp
+++ b/cpp/src/scheduler/ResourceMgr.cpp
@@ -105,6 +105,8 @@ void
 ResourceMgr::Clear() {
    std::lock_guard<std::mutex> lck(resources_mutex_);
    disk_resources_.clear();
+    cpu_resources_.clear();
+    gpu_resources_.clear();
    resources_.clear();
 }

--- a/cpp/unittest/db/utils.cpp
+++ b/cpp/unittest/db/utils.cpp
@@ -123,6 +123,7 @@ DBTest::TearDown() {
    ms::scheduler::JobMgrInst::GetInstance()->Stop();
    ms::scheduler::SchedInst::GetInstance()->Stop();
    ms::scheduler::ResMgrInst::GetInstance()->Stop();
+    ms::scheduler::ResMgrInst::GetInstance()->Clear();
    BaseTest::TearDown();