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(double free_space, double payload);

        double free_space;
    };

    class StorageServiceFileLookupRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFileLookupRequestMessage(std::string answer_mailbox, WorkflowFile *file, std::string& dst_partition, double payload);

        std::string answer_mailbox;
        WorkflowFile *file;
        std::string dst_partition;
    };

    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::string& dst_partition,
                                               double payload);

        std::string answer_mailbox;
        WorkflowFile *file;
        std::string dst_partition;
    };

    class StorageServiceFileDeleteAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileDeleteAnswerMessage(WorkflowFile *file,
                                              StorageService *storage_service,
                                              bool success,
                                              std::shared_ptr<FailureCause> failure_cause,
                                              double payload);

        WorkflowFile *file;
        StorageService *storage_service;
        bool success;
        std::shared_ptr<FailureCause> failure_cause;
    };

    class StorageServiceFileCopyRequestMessage : public StorageServiceMessage {
    public:
        StorageServiceFileCopyRequestMessage(std::string answer_mailbox, WorkflowFile *file, StorageService *src,
                                             std::string& src_partition, StorageService * dst, std::string& dst_partition,
                                             FileRegistryService *file_registry_service,
                                             SimulationTimestampFileCopyStart *start_timestamp,
                                             double payload);

        std::string answer_mailbox;
        WorkflowFile *file;
        StorageService *src;
        std::string src_partition;
        StorageService *dst;
        std::string dst_partition;
        FileRegistryService *file_registry_service;
        SimulationTimestampFileCopyStart *start_timestamp;
    };

    class StorageServiceFileCopyAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileCopyAnswerMessage(WorkflowFile *file, StorageService *storage_service,
                                            std::string dst_partition,
                                            FileRegistryService *file_registry_service,
                                            bool file_registry_service_updated,
                                            bool success, std::shared_ptr<FailureCause> cause,
                                            double payload);

        WorkflowFile *file;
        StorageService *storage_service;
        std::string dst_partition;
        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::string& dst_partition, double payload);

        std::string answer_mailbox;
        WorkflowFile *file;
        std::string dst_partition;
    };

    class StorageServiceFileWriteAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileWriteAnswerMessage(WorkflowFile *file,
                                             StorageService *storage_service,
                                             bool success,
                                             std::shared_ptr<FailureCause> failure_cause,
                                             std::string data_write_mailbox_name,
                                             double payload);

        WorkflowFile *file;
        StorageService *storage_service;
        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::string& src_partition, double payload);

        std::string answer_mailbox;
        std::string mailbox_to_receive_the_file_content;
        WorkflowFile *file;
        std::string src_partition;
    };

    class StorageServiceFileReadAnswerMessage : public StorageServiceMessage {
    public:
        StorageServiceFileReadAnswerMessage(WorkflowFile *file,
                                            StorageService *storage_service,
                                            bool success,
                                            std::shared_ptr<FailureCause> failure_cause,
                                            double payload);

        WorkflowFile *file;
        StorageService *storage_service;
        bool success;
        std::shared_ptr<FailureCause> failure_cause;
    };

    class StorageServiceFileContentMessage : public StorageServiceMessage {
    public:
        StorageServiceFileContentMessage(WorkflowFile *file);

        WorkflowFile *file;
    };

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

};


#endif //WRENCH_STORAGESERVICEMESSAGE_H