StorageServiceMessage.h

#ifndef WRENCH_STORAGESERVICEMESSAGE_H
#define WRENCH_STORAGESERVICEMESSAGE_H


#include <memory>

#include <wrench/services/ServiceMessage.h>
#include <wrench/workflow/execution_events/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,
                                             SimulationTimestampFileCopyStart *start_timestamp,
                                             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;
        SimulationTimestampFileCopyStart *start_timestamp;
    };

    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