StorageServiceMessage.h

 
#ifndef WRENCH_STORAGESERVICEMESSAGE_H
#define WRENCH_STORAGESERVICEMESSAGE_H
 
 
#include <memory>
 
#include "wrench/services/ServiceMessage.h"
#include "wrench/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(double payload);
    };
 
 
    class StorageServiceFreeSpaceRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFreeSpaceRequestMessage(simgrid::s4u::Mailbox *answer_mailbox, double payload);
 
        simgrid::s4u::Mailbox *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(simgrid::s4u::Mailbox *answer_mailbox, std::shared_ptr<DataFile>file,
                                               std::shared_ptr<FileLocation> location, double payload);
 
        simgrid::s4u::Mailbox *answer_mailbox;
        std::shared_ptr<DataFile>file;
        std::shared_ptr<FileLocation> location;
    };
 
    class StorageServiceFileLookupAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileLookupAnswerMessage(std::shared_ptr<DataFile>file, bool file_is_available, double payload);
 
        std::shared_ptr<DataFile>file;
        bool file_is_available;
    };
 
    class StorageServiceFileDeleteRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFileDeleteRequestMessage(simgrid::s4u::Mailbox *answer_mailbox,
                                               std::shared_ptr<DataFile>file,
                                               std::shared_ptr<FileLocation> location,
                                               double payload);
 
        simgrid::s4u::Mailbox *answer_mailbox;
        std::shared_ptr<DataFile>file;
        std::shared_ptr<FileLocation> location;
    };
 
    class StorageServiceFileDeleteAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileDeleteAnswerMessage(std::shared_ptr<DataFile>file,
                                              std::shared_ptr<StorageService> storage_service,
                                              bool success,
                                              std::shared_ptr<FailureCause> failure_cause,
                                              double payload);
 
        std::shared_ptr<DataFile>file;
        std::shared_ptr<StorageService> storage_service;
        bool success;
        std::shared_ptr<FailureCause> failure_cause;
    };
 
    class StorageServiceFileCopyRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFileCopyRequestMessage(simgrid::s4u::Mailbox *answer_mailbox,
                                             std::shared_ptr<DataFile> file,
                                             std::shared_ptr<FileLocation> src,
                                             std::shared_ptr<FileLocation> dst,
                                             std::shared_ptr<FileRegistryService> file_registry_service,
                                             double payload);
 
        simgrid::s4u::Mailbox *answer_mailbox;
        std::shared_ptr<DataFile>file;
        std::shared_ptr<FileLocation> src;
        std::shared_ptr<FileLocation> dst;
        std::shared_ptr<FileRegistryService> file_registry_service;
    };
 
    class StorageServiceFileCopyAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileCopyAnswerMessage(std::shared_ptr<DataFile>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);
 
        std::shared_ptr<DataFile>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(simgrid::s4u::Mailbox *answer_mailbox,
                                              std::shared_ptr<DataFile>file,
                                              std::shared_ptr<FileLocation> location,
                                              unsigned long buffer_size,
                                              double payload);
 
        simgrid::s4u::Mailbox *answer_mailbox;
        std::shared_ptr<DataFile>file;
        std::shared_ptr<FileLocation> location;
        unsigned long buffer_size;
    };
 
    class StorageServiceFileWriteAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileWriteAnswerMessage(std::shared_ptr<DataFile>file,
                                             std::shared_ptr<FileLocation> location,
                                             bool success,
                                             std::shared_ptr<FailureCause> failure_cause,
                                             simgrid::s4u::Mailbox *data_write_mailbox_name,
                                             double payload);
 
        std::shared_ptr<DataFile>file;
        std::shared_ptr<FileLocation> location;
        bool success;
        simgrid::s4u::Mailbox *data_write_mailbox;
        std::shared_ptr<FailureCause> failure_cause;
    };
 
    class StorageServiceFileReadRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFileReadRequestMessage(simgrid::s4u::Mailbox *answer_mailbox,
                                             simgrid::s4u::Mailbox *mailbox_to_receive_the_file_content,
                                             std::shared_ptr<DataFile>file,
                                             std::shared_ptr<FileLocation> location,
                                             double num_bytes_to_read,
                                             unsigned long buffer_size,
                                             double payload);
 
        simgrid::s4u::Mailbox *answer_mailbox;
        simgrid::s4u::Mailbox *mailbox_to_receive_the_file_content;
        std::shared_ptr<DataFile>file;
        std::shared_ptr<FileLocation> location;
        double num_bytes_to_read;
        unsigned long buffer_size;
    };
 
    class StorageServiceFileReadAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileReadAnswerMessage(std::shared_ptr<DataFile>file,
                                            std::shared_ptr<FileLocation> location,
                                            bool success,
                                            std::shared_ptr<FailureCause> failure_cause,
                                            double payload);
 
        std::shared_ptr<DataFile>file;
        std::shared_ptr<FileLocation> location;
        bool success;
        std::shared_ptr<FailureCause> failure_cause;
    };
 
    class StorageServiceFileContentChunkMessage : public StorageServiceMessage {
    public:
        explicit StorageServiceFileContentChunkMessage(std::shared_ptr<DataFile>file,
                unsigned long chunk_size, bool last_chunk);
 
        std::shared_ptr<DataFile>file;
        bool last_chunk;
    };
 
 
    class StorageServiceAckMessage : public StorageServiceMessage {
    public:
        StorageServiceAckMessage() : StorageServiceMessage(0) {}
    };
 
 
    /***********************/
    /***********************/
 
};
 
 
#endif //WRENCH_STORAGESERVICEMESSAGE_H