HTCondorCentralManagerService.h

#ifndef WRENCH_HTCONDORCENTRALMANAGERSERVICE_H
#define WRENCH_HTCONDORCENTRALMANAGERSERVICE_H

#include <set>
#include "wrench/services/compute/ComputeService.h"
#include "wrench/services/compute/htcondor/HTCondorCentralManagerServiceMessagePayload.h"

namespace wrench {

    class HTCondorCentralManagerService : public ComputeService {
    private:
        std::map<std::string, std::string> default_property_values = {};

        std::map<std::string, std::string> default_messagepayload_values = {
                {HTCondorCentralManagerServiceMessagePayload::STOP_DAEMON_MESSAGE_PAYLOAD,                  "1024"},
                {HTCondorCentralManagerServiceMessagePayload::DAEMON_STOPPED_MESSAGE_PAYLOAD,               "1024"},
                {HTCondorCentralManagerServiceMessagePayload::SUBMIT_STANDARD_JOB_REQUEST_MESSAGE_PAYLOAD,  "256000000"},
                {HTCondorCentralManagerServiceMessagePayload::SUBMIT_STANDARD_JOB_ANSWER_MESSAGE_PAYLOAD,   "256000000"},
                {HTCondorCentralManagerServiceMessagePayload::RESOURCE_DESCRIPTION_REQUEST_MESSAGE_PAYLOAD, "196000000"},
                {HTCondorCentralManagerServiceMessagePayload::RESOURCE_DESCRIPTION_ANSWER_MESSAGE_PAYLOAD,  "196000000"},
                {HTCondorCentralManagerServiceMessagePayload::STANDARD_JOB_DONE_MESSAGE_PAYLOAD,            "512000000"},
        };

    public:
        HTCondorCentralManagerService(const std::string &hostname,
                                      std::set<ComputeService *> compute_resources,
                                      std::map<std::string, std::string> property_list = {},
                                      std::map<std::string, std::string> messagepayload_list = {});

        /***********************/
        /***********************/

        void submitStandardJob(StandardJob *job,
                               std::map<std::string, std::string> &service_specific_arguments) override;

        void submitPilotJob(PilotJob *job, std::map<std::string, std::string> &service_specific_arguments) override;

        /***********************/
        /***********************/

        ~HTCondorCentralManagerService() override;

        void terminateStandardJob(StandardJob *job) override;

        void terminatePilotJob(PilotJob *job) override;

        /***********************/
        /***********************/

    private:
        int main() override;

        bool processNextMessage();

        void processSubmitStandardJob(const std::string &answer_mailbox, StandardJob *job,
                                      std::map<std::string, std::string> &service_specific_args);

        void processStandardJobCompletion(StandardJob *job);

        void processNegotiatorCompletion(std::vector<StandardJob *> pending_jobs);

        void terminate();

        std::set<ComputeService *> compute_resources;
        std::vector<StandardJob *> pending_jobs;
        bool dispatching_jobs = false;
        bool resources_unavailable = false;
        std::map<ComputeService *, unsigned long> compute_resources_map;
        std::map<StandardJob *, ComputeService *> running_jobs;
    };

}

#endif //WRENCH_HTCONDORCENTRALMANAGERSERVICE_H