DataMovementManager.h

 
#ifndef WRENCH_DATAMOVEMENTMANAGER_H
#define WRENCH_DATAMOVEMENTMANAGER_H
 
 
#include "wrench/simgrid_S4U_util/S4U_Daemon.h"
 
namespace wrench {
 
    class Workflow;
    class WorkflowFile;
    class StorageService;
    class WMS;
 
    /***********************/
    /***********************/
 
    class DataMovementManager : public Service {
 
    public:
 
        void stop();
 
        void kill();
 
        void initiateAsynchronousFileCopy(WorkflowFile *file,
                                          std::shared_ptr<FileLocation> src,
                                          std::shared_ptr<FileLocation> dst,
                                          std::shared_ptr<FileRegistryService> file_registry_service=nullptr);
 
        void doSynchronousFileCopy(WorkflowFile *file,
                                   std::shared_ptr<FileLocation> src,
                                   std::shared_ptr<FileLocation> dst,
                                   std::shared_ptr<FileRegistryService> file_registry_service=nullptr);
 
    protected:
 
        /***********************/
        /***********************/
 
 
        friend class WMS;
 
        explicit DataMovementManager(std::shared_ptr<WMS> wms);
 
        /***********************/
        /***********************/
 
    private:
 
        int main();
 
        std::shared_ptr<WMS> wms = nullptr;
 
        bool processNextMessage();
 
        struct CopyRequestSpecs {
            WorkflowFile *file;
            std::shared_ptr<FileLocation> src;
            std::shared_ptr<FileLocation> dst;
            std::shared_ptr<FileRegistryService> file_registry_service;
 
            CopyRequestSpecs(WorkflowFile *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