StorageService.h

 
#ifndef WRENCH_STORAGESERVICE_H
#define WRENCH_STORAGESERVICE_H
 
 
#include <string>
#include <set>
 
#include <wrench/services/Service.h>
#include <wrench/workflow/failure_causes/FailureCause.h>
#include <wrench/services/file_registry/FileRegistryService.h>
#include <wrench/workflow/job/StandardJob.h>
#include <wrench/services/storage/storage_helpers/LogicalFileSystem.h>
#include <wrench/services/storage/storage_helpers/FileLocation.h>
 
 
namespace wrench {
 
    class Simulation;
    class WorkflowFile;
    class FailureCause;
    class FileRegistryService;
 
    class StorageService : public Service {
 
    public:
 
 
        /***********************/
        /***********************/
 
        void stop() override;
 
        virtual std::map<std::string, double> getFreeSpace();
 
        virtual std::map<std::string, double> getTotalSpace();
 
        std::string getMountPoint();
        std::set<std::string> getMountPoints();
        bool hasMultipleMountPoints();
        bool hasMountPoint(std::string mp);
 
        static bool lookupFile(WorkflowFile *file, std::shared_ptr<FileLocation> location);
        static void deleteFile(WorkflowFile *file, std::shared_ptr<FileLocation> location,
                std::shared_ptr<FileRegistryService> file_registry_service = nullptr);
        static void readFile(WorkflowFile *file, std::shared_ptr<FileLocation> location);
        static void writeFile(WorkflowFile *file, std::shared_ptr<FileLocation> location);
 
 
        /***********************/
        /***********************/
        bool isScratch();
        void setScratch();
 
        static void copyFile(WorkflowFile *file,
                                     std::shared_ptr<FileLocation> src_location,
                                     std::shared_ptr<FileLocation> dst_location);
 
 
        static void initiateFileCopy(std::string answer_mailbox,
                                      WorkflowFile *file,
                                      std::shared_ptr<FileLocation> src_location,
                                      std::shared_ptr<FileLocation> dst_location);
 
        static void readFiles(std::map<WorkflowFile *, std::shared_ptr<FileLocation>> locations);
 
        static void writeFiles(std::map<WorkflowFile *, std::shared_ptr<FileLocation>> locations);
 
 
        StorageService(const std::string &hostname,
                       const std::set<std::string> mount_points,
                       const std::string &service_name,
                       const std::string &data_mailbox_name_prefix);
 
    protected:
 
        friend class Simulation;
        friend class FileRegistryService;
        friend class FileTransferThread;
 
        static void stageFile(WorkflowFile *file , std::shared_ptr<FileLocation> location);
 
        unsigned long buffer_size;
 
        std::map<std::string, std::unique_ptr<LogicalFileSystem>> file_systems;
 
        /***********************/
 
 
        /***********************/
 
    private:
 
        enum FileOperation {
            READ,
            WRITE,
        };
 
        static void writeOrReadFiles(FileOperation action,
                                     std::map<WorkflowFile *, std::shared_ptr<FileLocation>> locations);
 
        void stageFile(WorkflowFile *file , std::string mountpoint, std::string directory);
 
        bool is_stratch;
 
 
    };
 
    /***********************/
    /***********************/
 
};
 
 
#endif //WRENCH_STORAGESERVICE_H