HTCondorCentralManagerService.h

 
#ifndef WRENCH_HTCONDORCENTRALMANAGERSERVICE_H
#define WRENCH_HTCONDORCENTRALMANAGERSERVICE_H
 
#include <set>
#include <deque>
#include "wrench/services/compute/ComputeService.h"
#include "wrench/services/compute/htcondor/HTCondorCentralManagerServiceMessagePayload.h"
 
namespace wrench {
    /***********************/
    /***********************/
 
    class HTCondorCentralManagerService : public ComputeService {
    private:
        WRENCH_PROPERTY_COLLECTION_TYPE default_property_values = {};
 
WRENCH_MESSAGE_PAYLOADCOLLECTION_TYPE  default_messagepayload_values = {
                {HTCondorCentralManagerServiceMessagePayload::STOP_DAEMON_MESSAGE_PAYLOAD,                  1024},
                {HTCondorCentralManagerServiceMessagePayload::DAEMON_STOPPED_MESSAGE_PAYLOAD,               1024},
                {HTCondorCentralManagerServiceMessagePayload::SUBMIT_STANDARD_JOB_REQUEST_MESSAGE_PAYLOAD,  1024},
                {HTCondorCentralManagerServiceMessagePayload::SUBMIT_STANDARD_JOB_ANSWER_MESSAGE_PAYLOAD,   1024},
                {HTCondorCentralManagerServiceMessagePayload::SUBMIT_COMPOUND_JOB_REQUEST_MESSAGE_PAYLOAD,  1024},
                {HTCondorCentralManagerServiceMessagePayload::SUBMIT_COMPOUND_JOB_ANSWER_MESSAGE_PAYLOAD,   1024},
                {HTCondorCentralManagerServiceMessagePayload::SUBMIT_PILOT_JOB_REQUEST_MESSAGE_PAYLOAD,     1024},
                {HTCondorCentralManagerServiceMessagePayload::SUBMIT_PILOT_JOB_ANSWER_MESSAGE_PAYLOAD,      1024},
                {HTCondorCentralManagerServiceMessagePayload::RESOURCE_DESCRIPTION_REQUEST_MESSAGE_PAYLOAD, 1024},
                {HTCondorCentralManagerServiceMessagePayload::RESOURCE_DESCRIPTION_ANSWER_MESSAGE_PAYLOAD,  1024},
                {HTCondorCentralManagerServiceMessagePayload::STANDARD_JOB_DONE_MESSAGE_PAYLOAD,            1024},
                {HTCondorCentralManagerServiceMessagePayload::COMPOUND_JOB_DONE_MESSAGE_PAYLOAD,            1024},
                {HTCondorCentralManagerServiceMessagePayload::PILOT_JOB_STARTED_MESSAGE_PAYLOAD,            1024},
                {HTCondorCentralManagerServiceMessagePayload::PILOT_JOB_EXPIRED_MESSAGE_PAYLOAD,            1024}
        };
 
    public:
 
 
        HTCondorCentralManagerService(const std::string &hostname,
                                      double negotiator_startup_overhead,
                                      std::set<std::shared_ptr<ComputeService>> compute_services,
                                      WRENCH_PROPERTY_COLLECTION_TYPE property_list = {},
                                      WRENCH_MESSAGE_PAYLOADCOLLECTION_TYPE messagepayload_list = {});
 
        virtual bool supportsStandardJobs() override;
        virtual bool supportsCompoundJobs() override;
        virtual bool supportsPilotJobs() override;
 
        void addComputeService(std::shared_ptr<ComputeService> compute_service);
 
//        void submitStandardJob(std::shared_ptr<StandardJob> job,
//                               const std::map<std::string, std::string> &service_specific_arguments);
 
        void submitCompoundJob(std::shared_ptr<CompoundJob> job,
                               const std::map<std::string, std::string> &service_specific_arguments) override;
 
//        void submitPilotJob(std::shared_ptr<PilotJob> job, const std::map<std::string, std::string> &service_specific_arguments) override;
 
 
        ~HTCondorCentralManagerService() override;
 
//        void terminateStandardJob(std::shared_ptr<StandardJob> job) override;
 
        void terminateCompoundJob(std::shared_ptr<CompoundJob> job) override { }  ;
 
//        void terminatePilotJob(std::shared_ptr<PilotJob> job) override;
 
        bool jobKindIsSupported(const std::shared_ptr<Job>& job, std::map<std::string, std::string> service_specific_arguments);
 
        bool jobCanRunSomewhere(std::shared_ptr<CompoundJob> job, std::map<std::string, std::string> service_specific_arguments);
 
    private:
 
        friend class HTCondorComputeService;
 
        int main() override;
 
        bool processNextMessage();
 
        void processSubmitCompoundJob(simgrid::s4u::Mailbox *answer_mailbox, std::shared_ptr<CompoundJob> job,
                                      std::map<std::string, std::string> &service_specific_args);
 
        void processSubmitPilotJob(simgrid::s4u::Mailbox *answer_mailbox, std::shared_ptr<PilotJob> job,
                                   std::map<std::string, std::string> &service_specific_args);
 
        void processPilotJobStarted(std::shared_ptr<PilotJob>job);
 
        void processPilotJobCompletion(std::shared_ptr<PilotJob>job);
 
        void processCompoundJobCompletion(std::shared_ptr<CompoundJob>job);
 
        void processNegotiatorCompletion(std::vector<std::shared_ptr<Job>> &pending_jobs);
 
        void terminate();
 
 
        std::set<std::shared_ptr<ComputeService>> compute_services;
        std::vector<std::tuple<std::shared_ptr<CompoundJob>, std::map<std::string, std::string>>> pending_jobs;
        std::map<std::shared_ptr<CompoundJob>, std::shared_ptr<ComputeService>> running_jobs;
        bool dispatching_jobs = false;
        bool resources_unavailable = false;
        double negotiator_startup_overhead = 0.0;
 
    };
 
    /***********************/
    /***********************/
 
}
 
#endif //WRENCH_HTCONDORCENTRALMANAGERSERVICE_H