StorageServiceMessage.h

 
#ifndef WRENCH_STORAGESERVICEMESSAGE_H
#define WRENCH_STORAGESERVICEMESSAGE_H
 
 
#include <memory>
 
#include <wrench/services/ServiceMessage.h>
#include <wrench/workflow/failure_causes/FailureCause.h>
#include <wrench/services/file_registry/FileRegistryService.h>
#include <wrench/simulation/SimulationTimestampTypes.h>
#include <wrench/simulation/Simulation.h>
#include <wrench/simulation/SimulationOutput.h>
 
namespace wrench {
 
    /***********************/
    /***********************/
 
    class StorageServiceMessage : public ServiceMessage {
    protected:
        StorageServiceMessage(std::string name, double payload);
    };
 
 
    class StorageServiceFreeSpaceRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFreeSpaceRequestMessage(std::string answer_mailbox, double payload);
 
        std::string answer_mailbox;
    };
 
    class StorageServiceFreeSpaceAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFreeSpaceAnswerMessage(std::map<std::string, double> free_space, double payload);
 
        std::map<std::string, double> free_space;
    };
 
    class StorageServiceFileLookupRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFileLookupRequestMessage(std::string answer_mailbox, WorkflowFile *file,
                                               std::shared_ptr<FileLocation> location, double payload);
 
        std::string answer_mailbox;
        WorkflowFile *file;
        std::shared_ptr<FileLocation> location;
    };
 
    class StorageServiceFileLookupAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileLookupAnswerMessage(WorkflowFile *file, bool file_is_available, double payload);
 
        WorkflowFile *file;
        bool file_is_available;
    };
 
    class StorageServiceFileDeleteRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFileDeleteRequestMessage(std::string answer_mailbox,
                                               WorkflowFile *file,
                                               std::shared_ptr<FileLocation> location,
                                               double payload);
 
        std::string answer_mailbox;
        WorkflowFile *file;
        std::shared_ptr<FileLocation> location;
    };
 
    class StorageServiceFileDeleteAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileDeleteAnswerMessage(WorkflowFile *file,
                                              std::shared_ptr<StorageService> storage_service,
                                              bool success,
                                              std::shared_ptr<FailureCause> failure_cause,
                                              double payload);
 
        WorkflowFile *file;
        std::shared_ptr<StorageService> storage_service;
        bool success;
        std::shared_ptr<FailureCause> failure_cause;
    };
 
    class StorageServiceFileCopyRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFileCopyRequestMessage(std::string answer_mailbox, WorkflowFile *file,
                                             std::shared_ptr<FileLocation> src,
                                             std::shared_ptr<FileLocation> dst,
                                             std::shared_ptr<FileRegistryService> file_registry_service,
                                             double payload);
 
        std::string answer_mailbox;
        WorkflowFile *file;
        std::shared_ptr<FileLocation> src;
        std::shared_ptr<FileLocation> dst;
        std::shared_ptr<FileRegistryService> file_registry_service;
    };
 
    class StorageServiceFileCopyAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileCopyAnswerMessage(WorkflowFile *file,
                                            std::shared_ptr<FileLocation> src,
                                            std::shared_ptr<FileLocation> dst,
                                            std::shared_ptr<FileRegistryService> file_registry_service,
                                            bool file_registry_service_updated,
                                            bool success, std::shared_ptr<FailureCause> cause,
                                            double payload);
 
        WorkflowFile *file;
        std::shared_ptr<FileLocation> src;
        std::shared_ptr<FileLocation> dst;
        std::shared_ptr<FileRegistryService> file_registry_service;
        bool file_registry_service_updated;
        bool success;
        std::shared_ptr<FailureCause> failure_cause;
 
    };
 
 
 
    class StorageServiceFileWriteRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFileWriteRequestMessage(std::string answer_mailbox,
                                              WorkflowFile *file,
                                              std::shared_ptr<FileLocation> location,
                                              unsigned long buffer_size,
                                              double payload);
 
        std::string answer_mailbox;
        WorkflowFile *file;
        std::shared_ptr<FileLocation> location;
        unsigned long buffer_size;
    };
 
    class StorageServiceFileWriteAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileWriteAnswerMessage(WorkflowFile *file,
                                             std::shared_ptr<FileLocation> location,
                                             bool success,
                                             std::shared_ptr<FailureCause> failure_cause,
                                             std::string data_write_mailbox_name,
                                             double payload);
 
        WorkflowFile *file;
        std::shared_ptr<FileLocation> location;
        bool success;
        std::string data_write_mailbox_name;
        std::shared_ptr<FailureCause> failure_cause;
    };
 
    class StorageServiceFileReadRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFileReadRequestMessage(std::string answer_mailbox,
                                             std::string mailbox_to_receive_the_file_content,
                                             WorkflowFile *file,
                                             std::shared_ptr<FileLocation> location,
                                             unsigned long buffer_size,
                                             double payload);
 
        std::string answer_mailbox;
        std::string mailbox_to_receive_the_file_content;
        WorkflowFile *file;
       std::shared_ptr<FileLocation> location;
        unsigned long buffer_size;
    };
 
    class StorageServiceFileReadAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileReadAnswerMessage(WorkflowFile *file,
                                            std::shared_ptr<FileLocation> location,
                                            bool success,
                                            std::shared_ptr<FailureCause> failure_cause,
                                            double payload);
 
        WorkflowFile *file;
        std::shared_ptr<FileLocation> location;
        bool success;
        std::shared_ptr<FailureCause> failure_cause;
    };
 
    class StorageServiceFileContentChunkMessage : public StorageServiceMessage {
    public:
        explicit StorageServiceFileContentChunkMessage(WorkflowFile *file,
                unsigned long chunk_size, bool last_chunk);
 
        WorkflowFile *file;
        bool last_chunk;
    };
 
 
    class StorageServiceAckMessage : public StorageServiceMessage {
    public:
        StorageServiceAckMessage() : StorageServiceMessage("ACK",0) {}
    };
 
 
    /***********************/
    /***********************/
 
};
 
 
#endif //WRENCH_STORAGESERVICEMESSAGE_H