Commit 7b5658a2 authored by 王翔宇's avatar 王翔宇
Browse files

Improvement dump function in scheduler 


Former-commit-id: b58ed363de07b55e4c06f1c0ac140f8e05df7f6c
parent 2bf5973e

core/src/scheduler/JobMgr.cpp

+9 −0
Original line number Diff line number Diff line
@@ -49,6 +49,15 @@ JobMgr::Stop() { 
    }

}



json

JobMgr::Dump() const {

    json ret{

        {"running", running_},

        {"event_queue_length", queue_.size()},

    };

    return ret;

}



void

JobMgr::Put(const JobPtr& job) {

    {

core/src/scheduler/JobMgr.h

+5 −1
Original line number Diff line number Diff line
@@ -28,13 +28,14 @@ 
#include <vector>



#include "ResourceMgr.h"

#include "interface/interfaces.h"

#include "job/Job.h"

#include "task/Task.h"



namespace milvus {

namespace scheduler {



class JobMgr {

class JobMgr : public interface::dumpable {

 public:

    explicit JobMgr(ResourceMgrPtr res_mgr);
@@ -44,6 +45,9 @@ class JobMgr { 
    void

    Stop();



    json

    Dump() const override;



 public:

    void

    Put(const JobPtr& job);

core/src/scheduler/ResourceMgr.cpp

+16 −12
Original line number Diff line number Diff line
@@ -170,16 +170,20 @@ ResourceMgr::GetNumGpuResource() const { 
    return num;

}



std::string

ResourceMgr::Dump() {

    std::stringstream ss;

    ss << "ResourceMgr contains " << resources_.size() << " resources." << std::endl;



json

ResourceMgr::Dump() const {

    json resources{};

    for (auto &res : resources_) {

        ss << res->Dump();

    }



    return ss.str();

        resources.push_back(res->Dump());

    }

    json ret{

        {"number_of_resource", resources_.size()},

        {"number_of_disk_resource", disk_resources_.size()},

        {"number_of_cpu_resource", cpu_resources_.size()},

        {"number_of_gpu_resource", gpu_resources_.size()},

        {"resources", resources},

    };

    return ret;

}



std::string
@@ -187,9 +191,9 @@ ResourceMgr::DumpTaskTables() { 
    std::stringstream ss;

    ss << ">>>>>>>>>>>>>>>ResourceMgr::DumpTaskTable<<<<<<<<<<<<<<<" << std::endl;

    for (auto& resource : resources_) {

        ss << resource->Dump() << std::endl;

        ss << resource->task_table().Dump();

        ss << resource->Dump() << std::endl << std::endl;

        ss << resource->name() << std::endl;

        ss << resource->task_table().Dump().dump();

        ss << resource->name() << std::endl << std::endl;

    }

    return ss.str();

}

core/src/scheduler/ResourceMgr.h

+4 −3
Original line number Diff line number Diff line
@@ -25,13 +25,14 @@ 
#include <utility>

#include <vector>



#include "interface/interfaces.h"

#include "resource/Resource.h"

#include "utils/Log.h"



namespace milvus {

namespace scheduler {



class ResourceMgr {

class ResourceMgr : public interface::dumpable {

 public:

    ResourceMgr() = default;
@@ -103,8 +104,8 @@ class ResourceMgr { 


 public:

    /******** Utility Functions ********/

    std::string

    Dump();

    json

    Dump() const override;



    std::string

    DumpTaskTables();

core/src/scheduler/Scheduler.cpp

+7 −3
Original line number Diff line number Diff line
@@ -66,9 +66,13 @@ Scheduler::PostEvent(const EventPtr& event) { 
    event_cv_.notify_one();

}



std::string

Scheduler::Dump() {

    return std::string();

json

Scheduler::Dump() const {

    json ret{

        {"running", running_},

        {"event_queue_length", event_queue_.size()},

    };

    return ret;

}



void