FileTransferThread.h

#ifndef WRENCH_FILETRANSFERTHREAD_H
#define WRENCH_FILETRANSFERTHREAD_H

#include <string>
#include <wrench/simgrid_S4U_util/S4U_Daemon.h>
#include <wrench/services/Service.h>
#include <wrench/services/storage/StorageService.h>

namespace wrench {

    class WorkflowFile;
    class SimulationTimestampFileCopyStart;

    class FileTransferThread : public Service {

    public:



        FileTransferThread(std::string hostname,
                           std::shared_ptr<StorageService> parent,
                           WorkflowFile *file,
                           std::string src_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,
                           unsigned long buffer_size,
                           SimulationTimestampFileCopyStart *start_timestamp = nullptr);

        FileTransferThread(std::string hostname,
                           std::shared_ptr<StorageService> parent,
                           WorkflowFile *file,
                           std::shared_ptr<FileLocation> src_location,
                           std::string dst_mailbox,
                           std::string answer_mailbox_if_read,
                           std::string answer_mailbox_if_write,
                           std::string answer_mailbox_if_copy,
                           unsigned long buffer_size,
                           SimulationTimestampFileCopyStart *start_timestamp = nullptr);

        FileTransferThread(std::string hostname,
                           std::shared_ptr<StorageService> parent,
                           WorkflowFile *file,
                           std::shared_ptr<FileLocation> src_location,
                           std::shared_ptr<FileLocation> dsg_location,
                           std::string answer_mailbox_if_read,
                           std::string answer_mailbox_if_write,
                           std::string answer_mailbox_if_copy,
                           unsigned long buffer_size,
                           SimulationTimestampFileCopyStart *start_timestamp = nullptr);

        int main() override;
        void cleanup(bool has_returned_from_main, int return_value) override;


    private:

        std::shared_ptr<StorageService> parent;
        WorkflowFile *file;

        // Only one of these two is valid
        std::string src_mailbox;
        std::shared_ptr<FileLocation> src_location;

        // Only one of these two is valid
        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;
        unsigned long buffer_size;
        SimulationTimestampFileCopyStart *start_timestamp;

        void receiveFileFromNetwork(WorkflowFile *file, std::string mailbox, std::shared_ptr<FileLocation> location);
        void sendLocalFileToNetwork(WorkflowFile *file, std::shared_ptr<FileLocation> location, std::string mailbox);
        void downloadFileFromStorageService(WorkflowFile *file, std::shared_ptr<FileLocation> src_location, std::shared_ptr<FileLocation> dst_location);
        void copyFileLocally(WorkflowFile *file, std::shared_ptr<FileLocation> src_location, std::shared_ptr<FileLocation> dst_location);

    };

}

#endif //WRENCH_FILETRANSFERTHREAD_H