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, StorageService *src, StorageService *dst, FileRegistryService *file_registry_service=nullptr);
        void initiateAsynchronousFileCopy(WorkflowFile *file,
                                          StorageService *src, std::string src_partition,
                                          StorageService *dst, std::string dst_partition,
                                          FileRegistryService *file_registry_service=nullptr);

        void doSynchronousFileCopy(WorkflowFile *file, StorageService *src, StorageService *dst, FileRegistryService *file_registry_service=nullptr);
        void doSynchronousFileCopy(WorkflowFile *file,
                                   StorageService *src, std::string src_partition,
                                   StorageService *dst, std::string dst_partition,
                                   FileRegistryService *file_registry_service=nullptr);

    protected:

        /***********************/
        /***********************/


        friend class WMS;

        explicit DataMovementManager(WMS *wms);

        /***********************/
        /***********************/

    private:

        int main();

        WMS *wms = nullptr;

        bool processNextMessage();

        struct CopyRequestSpecs {
            WorkflowFile *file;
            StorageService *dst;
            std::string dst_partition;
            FileRegistryService *file_registry_service;

            CopyRequestSpecs(WorkflowFile *file,
            StorageService *dst, std::string dst_partition,
            FileRegistryService *file_registry_service) : file(file), dst(dst), dst_partition(dst_partition), file_registry_service(file_registry_service) {}

            bool operator==(const CopyRequestSpecs &rhs) const {
              return (file == rhs.file) && (dst == rhs.dst) && (dst_partition == rhs.dst_partition);
            }
        };

        std::list<std::unique_ptr<CopyRequestSpecs>> pending_file_copies;
    };

    /***********************/
    /***********************/


};


#endif //WRENCH_DATAMOVEMENTMANAGER_H