CloudService.h

#ifndef WRENCH_CLOUDSERVICE_H
#define WRENCH_CLOUDSERVICE_H

#include <map>
#include <simgrid/s4u/VirtualMachine.hpp>

#include "wrench/services/compute/ComputeServiceMessage.h"
#include "wrench/services/compute/ComputeService.h"
#include "wrench/services/compute/cloud/CloudServiceProperty.h"
#include "wrench/services/compute/cloud/CloudServiceMessagePayload.h"
#include "wrench/simulation/Simulation.h"
#include "wrench/simgrid_S4U_util/S4U_VirtualMachine.h"
#include "wrench/workflow/job/PilotJob.h"


namespace wrench {

    class Simulation;

    class ComputeService;

    class CloudService : public ComputeService {

    private:
        std::map<std::string, std::string> default_property_values = {
                {CloudServiceProperty::SUPPORTS_PILOT_JOBS,         "false"},
                {CloudServiceProperty::SUPPORTS_STANDARD_JOBS,      "true"},
                {CloudServiceProperty::VM_BOOT_OVERHEAD_IN_SECONDS, "0.0"}
        };

        std::map<std::string, std::string> default_messagepayload_values = {
                {CloudServiceMessagePayload::STOP_DAEMON_MESSAGE_PAYLOAD,                  "1024"},
                {CloudServiceMessagePayload::DAEMON_STOPPED_MESSAGE_PAYLOAD,               "1024"},
                {CloudServiceMessagePayload::RESOURCE_DESCRIPTION_REQUEST_MESSAGE_PAYLOAD, "1024"},
                {CloudServiceMessagePayload::RESOURCE_DESCRIPTION_ANSWER_MESSAGE_PAYLOAD,  "1024"},
                {CloudServiceMessagePayload::GET_EXECUTION_HOSTS_REQUEST_MESSAGE_PAYLOAD,  "1024"},
                {CloudServiceMessagePayload::GET_EXECUTION_HOSTS_ANSWER_MESSAGE_PAYLOAD,   "1024"},
                {CloudServiceMessagePayload::CREATE_VM_REQUEST_MESSAGE_PAYLOAD,            "1024"},
                {CloudServiceMessagePayload::CREATE_VM_ANSWER_MESSAGE_PAYLOAD,             "1024"},
                {CloudServiceMessagePayload::SHUTDOWN_VM_REQUEST_MESSAGE_PAYLOAD,          "1024"},
                {CloudServiceMessagePayload::SHUTDOWN_VM_ANSWER_MESSAGE_PAYLOAD,           "1024"},
                {CloudServiceMessagePayload::START_VM_REQUEST_MESSAGE_PAYLOAD,             "1024"},
                {CloudServiceMessagePayload::START_VM_ANSWER_MESSAGE_PAYLOAD,              "1024"},
                {CloudServiceMessagePayload::SUSPEND_VM_REQUEST_MESSAGE_PAYLOAD,           "1024"},
                {CloudServiceMessagePayload::SUSPEND_VM_ANSWER_MESSAGE_PAYLOAD,            "1024"},
                {CloudServiceMessagePayload::RESUME_VM_REQUEST_MESSAGE_PAYLOAD,            "1024"},
                {CloudServiceMessagePayload::RESUME_VM_ANSWER_MESSAGE_PAYLOAD,             "1024"},
                {CloudServiceMessagePayload::SUBMIT_STANDARD_JOB_REQUEST_MESSAGE_PAYLOAD,  "1024"},
                {CloudServiceMessagePayload::SUBMIT_STANDARD_JOB_ANSWER_MESSAGE_PAYLOAD,   "1024"},
                {CloudServiceMessagePayload::SUBMIT_PILOT_JOB_REQUEST_MESSAGE_PAYLOAD,     "1024"},
                {CloudServiceMessagePayload::SUBMIT_PILOT_JOB_ANSWER_MESSAGE_PAYLOAD,      "1024"}
        };

    public:
        CloudService(const std::string &hostname,
                     std::vector<std::string> &execution_hosts,
                     double scratch_space_size,
                     std::map<std::string, std::string> property_list = {},
                     std::map<std::string, std::string> messagepayload_list = {});

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

        virtual std::string createVM(unsigned long num_cores = ComputeService::ALL_CORES,
                                     double ram_memory = ComputeService::ALL_RAM,
                                     std::map<std::string, std::string> property_list = {},
                                     std::map<std::string, std::string> messagepayload_list = {});

        virtual bool shutdownVM(const std::string &vm_hostname);

        virtual bool startVM(const std::string &vm_hostname);

        virtual bool suspendVM(const std::string &vm_hostname);

        virtual bool resumeVM(const std::string &vm_hostname);

        std::vector<std::string> getExecutionHosts();

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

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

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

        void terminateStandardJob(StandardJob *job) override;
        void terminatePilotJob(PilotJob *job) override;

        void validateProperties();

        ~CloudService();

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

    protected:
        /***********************/
        /***********************/

        friend class Simulation;

        int main() override;

        std::unique_ptr<SimulationMessage> sendRequest(std::string &answer_mailbox, ComputeServiceMessage *message);

        virtual bool processNextMessage();

        virtual void processGetResourceInformation(const std::string &answer_mailbox);

        virtual void processGetExecutionHosts(const std::string &answer_mailbox);

        virtual void processCreateVM(const std::string &answer_mailbox,
                                     const std::string &pm_hostname,
                                     const std::string &vm_name,
                                     unsigned long num_cores,
                                     double ram_memory,
                                     std::map<std::string, std::string> &property_list,
                                     std::map<std::string, std::string> &messagepayload_list);

        virtual void processShutdownVM(const std::string &answer_mailbox, const std::string &vm_hostname);

        virtual void processStartVM(const std::string &answer_mailbox, const std::string &vm_name);

        virtual void processSuspendVM(const std::string &answer_mailbox, const std::string &vm_hostname);

        virtual void processResumeVM(const std::string &answer_mailbox, const std::string &vm_hostname);

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

        virtual void processSubmitPilotJob(const std::string &answer_mailbox, PilotJob *job,
                                           std::map<std::string, std::string> &service_specific_args);

        void stopAllVMs();

        static unsigned long VM_ID;
        std::vector<std::string> execution_hosts;

        std::map<std::string, double> cs_available_ram;

        std::map<std::string, std::tuple<std::shared_ptr<S4U_VirtualMachine>, std::shared_ptr<ComputeService>, unsigned long, unsigned long>> vm_list;

        std::map<std::string, unsigned long> used_cores_per_execution_host;

        /***********************/
        /***********************/
    };
}

#endif //WRENCH_CLOUDSERVICE_H