VirtualizedClusterComputeService.h

 
#ifndef WRENCH_VIRTUALIZEDCLUSTERSERVICE_H
#define WRENCH_VIRTUALIZEDCLUSTERSERVICE_H
 
#include "wrench/services/compute/virtualized_cluster/VirtualizedClusterComputeServiceProperty.h"
#include "wrench/services/compute/virtualized_cluster/VirtualizedClusterComputeServiceMessagePayload.h"
#include <wrench/services/compute/cloud/CloudComputeService.h>
 
namespace wrench {
 
    class Simulation;
 
    class ComputeService;
 
    class CloudComputeService;
 
    class VirtualizedClusterComputeService : public CloudComputeService {
    private:
        std::map<std::string, std::string> default_property_values = {
                {VirtualizedClusterComputeServiceProperty::VM_BOOT_OVERHEAD_IN_SECONDS, "0.0"}
        };
 
        std::map<std::string, double> default_messagepayload_values = {
                {VirtualizedClusterComputeServiceMessagePayload::STOP_DAEMON_MESSAGE_PAYLOAD,                  1024},
                {VirtualizedClusterComputeServiceMessagePayload::DAEMON_STOPPED_MESSAGE_PAYLOAD,               1024},
                {VirtualizedClusterComputeServiceMessagePayload::RESOURCE_DESCRIPTION_REQUEST_MESSAGE_PAYLOAD, 1024},
                {VirtualizedClusterComputeServiceMessagePayload::RESOURCE_DESCRIPTION_ANSWER_MESSAGE_PAYLOAD,  1024},
                {VirtualizedClusterComputeServiceMessagePayload::GET_EXECUTION_HOSTS_REQUEST_MESSAGE_PAYLOAD,  1024},
                {VirtualizedClusterComputeServiceMessagePayload::GET_EXECUTION_HOSTS_ANSWER_MESSAGE_PAYLOAD,   1024},
                {VirtualizedClusterComputeServiceMessagePayload::CREATE_VM_REQUEST_MESSAGE_PAYLOAD,            1024},
                {VirtualizedClusterComputeServiceMessagePayload::CREATE_VM_ANSWER_MESSAGE_PAYLOAD,             1024},
                {VirtualizedClusterComputeServiceMessagePayload::MIGRATE_VM_REQUEST_MESSAGE_PAYLOAD,           1024},
                {VirtualizedClusterComputeServiceMessagePayload::MIGRATE_VM_ANSWER_MESSAGE_PAYLOAD,            1024},
                {VirtualizedClusterComputeServiceMessagePayload::SUBMIT_STANDARD_JOB_REQUEST_MESSAGE_PAYLOAD,  1024},
                {VirtualizedClusterComputeServiceMessagePayload::SUBMIT_STANDARD_JOB_ANSWER_MESSAGE_PAYLOAD,   1024},
                {VirtualizedClusterComputeServiceMessagePayload::SUBMIT_PILOT_JOB_REQUEST_MESSAGE_PAYLOAD,     1024},
                {VirtualizedClusterComputeServiceMessagePayload::SUBMIT_PILOT_JOB_ANSWER_MESSAGE_PAYLOAD,      1024},
                {VirtualizedClusterComputeServiceMessagePayload::IS_THERE_AT_LEAST_ONE_HOST_WITH_AVAILABLE_RESOURCES_REQUEST_MESSAGE_PAYLOAD, 1024},
                {VirtualizedClusterComputeServiceMessagePayload::IS_THERE_AT_LEAST_ONE_HOST_WITH_AVAILABLE_RESOURCES_ANSWER_MESSAGE_PAYLOAD, 1024},
        };
 
    public:
        VirtualizedClusterComputeService(const std::string &hostname,
                                         std::vector<std::string> &execution_hosts,
                                         std::string scratch_space_mount_point,
                                         std::map<std::string, std::string> property_list = {},
                                         std::map<std::string, double> messagepayload_list = {});
 
        /***********************/
        /***********************/
 
        using CloudComputeService::startVM;
 
        virtual std::shared_ptr<BareMetalComputeService>
        startVM(const std::string &vm_name, const std::string &pm_name);
 
        virtual void migrateVM(const std::string &vm_name, const std::string &dest_pm_hostname);
 
        /***********************/
        /***********************/
 
    protected:
 
        /***********************/
        /***********************/
 
        friend class Simulation;
 
        int main() override;
 
        bool processNextMessage() override;
 
        virtual void processMigrateVM(const std::string &answer_mailbox,
                                      const std::string &vm_name,
                                      const std::string &dest_pm_hostname);
 
        /***********************/
        /***********************/
    };
 
}
 
#endif //WRENCH_VIRTUALIZEDCLUSTERSERVICE_H