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);
 
        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);
 
        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);
 
        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;
 
        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