DataMovementManager.h

 
#ifndef WRENCH_DATAMOVEMENTMANAGER_H
#define WRENCH_DATAMOVEMENTMANAGER_H
 
#include <list>
 
#include "wrench/simgrid_S4U_util/S4U_Daemon.h"
#include "wrench/services/storage/storage_helpers/FileLocation.h"
 
namespace wrench {
 
    class Workflow;
    class DataFile;
    class StorageService;
    class FileRegistryService;
    class ExecutionController;
    class WMS;
 
    /***********************/
    /***********************/
 
    class DataMovementManager : public Service {
 
    public:
 
        void stop() override;
 
        void kill();
 
        void initiateAsynchronousFileCopy(std::shared_ptr<DataFile>file,
                                          std::shared_ptr<FileLocation> src,
                                          std::shared_ptr<FileLocation> dst,
                                          std::shared_ptr<FileRegistryService> file_registry_service=nullptr);
 
        void doSynchronousFileCopy(std::shared_ptr<DataFile>file,
                                   std::shared_ptr<FileLocation> src,
                                   std::shared_ptr<FileLocation> dst,
                                   std::shared_ptr<FileRegistryService> file_registry_service=nullptr);
 
 
    protected:
 
        /***********************/
        /***********************/
 
        friend class WMS;
        friend class ExecutionController;
 
        explicit DataMovementManager(std::string hostname, simgrid::s4u::Mailbox *creator_mailbox);
 
        /***********************/
        /***********************/
 
    private:
 
        int main() override;
 
        simgrid::s4u::Mailbox *getCreatorMailbox();
 
        simgrid::s4u::Mailbox *creator_mailbox;
 
        bool processNextMessage();
 
        struct CopyRequestSpecs {
            std::shared_ptr<DataFile>file;
            std::shared_ptr<FileLocation> src;
            std::shared_ptr<FileLocation> dst;
            std::shared_ptr<FileRegistryService> file_registry_service;
 
            ~CopyRequestSpecs() {
            }
 
            CopyRequestSpecs(std::shared_ptr<DataFile> file,
                             std::shared_ptr<FileLocation> src,
                             std::shared_ptr<FileLocation> dst,
                             std::shared_ptr<FileRegistryService> file_registry_service) :
                    file(file), src(src), dst(dst), file_registry_service(file_registry_service) {}
 
            bool operator==(const CopyRequestSpecs &rhs) const {
                return (file == rhs.file) &&
                       (dst->getStorageService() == rhs.dst->getStorageService())  &&
                       (dst->getAbsolutePathAtMountPoint()     == rhs.dst->getAbsolutePathAtMountPoint());
            }
        };
 
        std::list<std::unique_ptr<CopyRequestSpecs>> pending_file_copies;
    };
 
    /***********************/
    /***********************/
 
 
};
 
 
#endif //WRENCH_DATAMOVEMENTMANAGER_H