HTCondorNegotiatorService.h

 
#ifndef WRENCH_HTCONDORNEGOTIATORSERVICE_H
#define WRENCH_HTCONDORNEGOTIATORSERVICE_H
 
#include "wrench/services/Service.h"
#include "wrench/services/compute/ComputeService.h"
#include "wrench/services/compute/htcondor/HTCondorCentralManagerServiceMessagePayload.h"
#include "wrench/job/Job.h"
 
namespace wrench {
 
    /***********************/
    /***********************/
    class HTCondorNegotiatorService : public Service {
    private:
WRENCH_MESSAGE_PAYLOADCOLLECTION_TYPE  default_messagepayload_values = {
                {HTCondorCentralManagerServiceMessagePayload::STOP_DAEMON_MESSAGE_PAYLOAD,              1024},
                {HTCondorCentralManagerServiceMessagePayload::DAEMON_STOPPED_MESSAGE_PAYLOAD,           1024},
                {HTCondorCentralManagerServiceMessagePayload::HTCONDOR_NEGOTIATOR_DONE_MESSAGE_PAYLOAD, 1024},
        };
 
    public:
 
        HTCondorNegotiatorService(std::string &hostname,
                                  double startup_overhead,
                                  std::set<std::shared_ptr<ComputeService>> &compute_services,
                                  std::map<std::shared_ptr<CompoundJob>, std::shared_ptr<ComputeService>> &running_jobs,
                                  std::vector<std::tuple<std::shared_ptr<CompoundJob>, std::map<std::string, std::string>>> &pending_jobs,
                                  simgrid::s4u::Mailbox *reply_mailbox);
 
        ~HTCondorNegotiatorService();
 
    private:
        int main() override;
 
        struct JobPriorityComparator {
            bool operator()(std::tuple<std::shared_ptr<CompoundJob>, std::map<std::string, std::string>> &lhs,
                            std::tuple<std::shared_ptr<CompoundJob>, std::map<std::string, std::string>> &rhs);
        };
 
        std::shared_ptr<ComputeService> pickTargetComputeService(std::shared_ptr<CompoundJob> job, std::map<std::string, std::string> service_specific_arguments);
        std::shared_ptr<ComputeService> pickTargetComputeServiceGridUniverse(std::shared_ptr<CompoundJob> job, std::map<std::string, std::string> service_specific_arguments);
        std::shared_ptr<ComputeService> pickTargetComputeServiceNonGridUniverse(std::shared_ptr<CompoundJob> job, std::map<std::string, std::string> service_specific_arguments);
 
        double startup_overhead;
        simgrid::s4u::Mailbox *reply_mailbox;
        std::set<std::shared_ptr<ComputeService>> compute_services;
        std::map<std::shared_ptr<CompoundJob>, std::shared_ptr<ComputeService>> running_jobs;
        std::vector<std::tuple<std::shared_ptr<CompoundJob>, std::map<std::string, std::string>>> pending_jobs;
    };
 
    /***********************/
    /***********************/
}
 
#endif //WRENCH_HTCONDORNEGOTIATORSERVICE_H