MS-611 Add resources validity check in ResourceMgr

Former-commit-id: 8732bacae82613066c415d4e22db256f9b8056d2

MS-611 Add resources validity check in ResourceMgr
Former-commit-id: 8732bacae82613066c415d4e22db256f9b8056d2
6d0a2489 · wxyu · e27e6741 · 6d0a2489 · 6d0a2489
隐藏空白更改
内联并排

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();