FileTransferThreadMessage.h

#ifndef WRENCH_FILETRANSFERTHREADMESSAGE_H
#define WRENCH_FILETRANSFERTHREADMESSAGE_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>
#include "wrench/services/storage/storage_helpers/FileTransferThread.h"

namespace wrench {

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

    class FileTransferThreadMessage : public ServiceMessage {
    protected:
        FileTransferThreadMessage(std::string name, double payload) :
                ServiceMessage("FileTransferThreadMessage::" + name, payload) {}
    };


    class FileTransferThreadNotificationMessage : public FileTransferThreadMessage {
    public:
        FileTransferThreadNotificationMessage(std::shared_ptr<FileTransferThread> file_transfer_thread,
                                              WorkflowFile *file,
                                              std::string src_mailbox,
                                              std::shared_ptr<FileLocation> src_location,
                                              std::string dst_mailbox,
                                              std::shared_ptr<FileLocation> dst_location,
                                              std::string answer_mailbox_if_read,
                                              std::string answer_mailbox_if_write,
                                              std::string answer_mailbox_if_copy,
                                              bool success, std::shared_ptr<FailureCause> failure_cause,
                                              SimulationTimestampFileCopyStart *start_time_stamp) :
                FileTransferThreadMessage("FileTransferThreadNotificationMessage", 0),
                file_transfer_thread(file_transfer_thread),
                file(file),
                src_mailbox(src_mailbox), src_location(src_location),
                dst_mailbox(dst_mailbox), dst_location(dst_location),
                answer_mailbox_if_read(answer_mailbox_if_read),
                answer_mailbox_if_write(answer_mailbox_if_write),
                answer_mailbox_if_copy(answer_mailbox_if_copy),
                success(success),
                failure_cause(failure_cause), start_time_stamp(start_time_stamp) {}

        std::shared_ptr<FileTransferThread> file_transfer_thread;
        WorkflowFile *file;

        std::string src_mailbox;
        std::shared_ptr<FileLocation> src_location;

        std::string dst_mailbox;
        std::shared_ptr<FileLocation> dst_location;

        std::string answer_mailbox_if_read;
        std::string answer_mailbox_if_write;
        std::string answer_mailbox_if_copy;
        bool success;
        std::shared_ptr<FailureCause> failure_cause;
        SimulationTimestampFileCopyStart *start_time_stamp;
    };


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

};


#endif //WRENCH_FILETRANSFERTHREADMESSAGE_H