FileTransferThreadMessage.h

 
#ifndef WRENCH_FILETRANSFERTHREADMESSAGE_H
#define WRENCH_FILETRANSFERTHREADMESSAGE_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"
#include "wrench/services/storage/storage_helpers/FileTransferThread.h"
 
namespace wrench {
 
    /***********************/
    /***********************/
 
    class FileTransferThreadMessage : public ServiceMessage {
    protected:
        FileTransferThreadMessage(double payload) :
                ServiceMessage(payload) {}
    };
 
 
    class FileTransferThreadNotificationMessage : public FileTransferThreadMessage {
    public:
        FileTransferThreadNotificationMessage(std::shared_ptr<FileTransferThread> file_transfer_thread,
                                              std::shared_ptr<DataFile>file,
                                              simgrid::s4u::Mailbox *src_mailbox,
                                              std::shared_ptr<FileLocation> src_location,
                                              simgrid::s4u::Mailbox *dst_mailbox,
                                              std::shared_ptr<FileLocation> dst_location,
                                              simgrid::s4u::Mailbox *answer_mailbox_if_read,
                                              simgrid::s4u::Mailbox *answer_mailbox_if_write,
                                              simgrid::s4u::Mailbox *answer_mailbox_if_copy,
                                              bool success, std::shared_ptr<FailureCause> failure_cause) :
                FileTransferThreadMessage(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) {}
 
        std::shared_ptr<FileTransferThread> file_transfer_thread;
        std::shared_ptr<DataFile>file;
 
        simgrid::s4u::Mailbox *src_mailbox;
        std::shared_ptr<FileLocation> src_location;
 
        simgrid::s4u::Mailbox *dst_mailbox;
        std::shared_ptr<FileLocation> dst_location;
 
        simgrid::s4u::Mailbox *answer_mailbox_if_read;
        simgrid::s4u::Mailbox *answer_mailbox_if_write;
        simgrid::s4u::Mailbox *answer_mailbox_if_copy;
        bool success;
        std::shared_ptr<FailureCause> failure_cause;
    };
 
 
    /***********************/
    /***********************/
 
};
 
 
#endif //WRENCH_FILETRANSFERTHREADMESSAGE_H