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"
#include <iostream>
 
namespace wrench {
 
    class DataFile;
    class SimulationTimestampFileCopyStart;
 
    /***********************/
    /***********************/
 
    class FileTransferThread : public Service {
 
    public:
 
        ~FileTransferThread() {
        }
 
 
        FileTransferThread(std::string hostname,
                           std::shared_ptr<StorageService> parent,
                           std::shared_ptr<DataFile>file,
                           double num_bytes_to_transfer,
                           simgrid::s4u::Mailbox *src_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,
                           unsigned long buffer_size);
 
        FileTransferThread(std::string hostname,
                           std::shared_ptr<StorageService> parent,
                           std::shared_ptr<DataFile>file,
                           double num_bytes_to_transfer,
                           std::shared_ptr<FileLocation> src_location,
                           simgrid::s4u::Mailbox *dst_mailbox,
                           simgrid::s4u::Mailbox *answer_mailbox_if_read,
                           simgrid::s4u::Mailbox *answer_mailbox_if_write,
                           simgrid::s4u::Mailbox *answer_mailbox_if_copy,
                           unsigned long buffer_size);
 
        FileTransferThread(std::string hostname,
                           std::shared_ptr<StorageService> parent,
                           std::shared_ptr<DataFile>file,
                           double num_bytes_to_transfer,
                           std::shared_ptr<FileLocation> src_location,
                           std::shared_ptr<FileLocation> dsg_location,
                           simgrid::s4u::Mailbox *answer_mailbox_if_read,
                           simgrid::s4u::Mailbox *answer_mailbox_if_write,
                           simgrid::s4u::Mailbox *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;
        std::shared_ptr<DataFile>file;
 
        // Only one of these two is valid
        simgrid::s4u::Mailbox *src_mailbox;
        std::shared_ptr<FileLocation> src_location;
 
        // Only one of these two is valid
        simgrid::s4u::Mailbox *dst_mailbox;
        std::shared_ptr<FileLocation> dst_location;
 
        double num_bytes_to_transfer;
 
        simgrid::s4u::Mailbox *answer_mailbox_if_read;
        simgrid::s4u::Mailbox *answer_mailbox_if_write;
        simgrid::s4u::Mailbox *answer_mailbox_if_copy;
        unsigned long buffer_size;
 
        void receiveFileFromNetwork(std::shared_ptr<DataFile> file, simgrid::s4u::Mailbox *mailbox, std::shared_ptr<FileLocation> location);
        void sendLocalFileToNetwork(std::shared_ptr<DataFile> file, std::shared_ptr<FileLocation> location, double num_bytes_to_transfer, simgrid::s4u::Mailbox *mailbox);
        void downloadFileFromStorageService(std::shared_ptr<DataFile>file, std::shared_ptr<FileLocation> src_location, std::shared_ptr<FileLocation> dst_location);
        void copyFileLocally(std::shared_ptr<DataFile>file, std::shared_ptr<FileLocation> src_location, std::shared_ptr<FileLocation> dst_location);
 
    };
 
    /***********************/
    /***********************/
}
 
#endif //WRENCH_FILETRANSFERTHREAD_H