StorageService.h

 
#ifndef WRENCH_STORAGESERVICE_H
#define WRENCH_STORAGESERVICE_H
 
 
#include <string>
#include <set>
 
#include "wrench/services/Service.h"
#include "wrench/failure_causes/FailureCause.h"
#include "wrench/services/file_registry/FileRegistryService.h"
#include "wrench/job/StandardJob.h"
#include "wrench/services/storage/storage_helpers/LogicalFileSystem.h"
#include "wrench/services/storage/storage_helpers/FileLocation.h"
 
 
namespace wrench {
 
    class Simulation;
    class DataFile;
    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(std::shared_ptr<DataFile>file, std::shared_ptr<FileLocation> location);
        static void deleteFile(std::shared_ptr<DataFile>file, std::shared_ptr<FileLocation> location,
                std::shared_ptr<FileRegistryService> file_registry_service = nullptr);
        static void readFile(std::shared_ptr<DataFile>file, std::shared_ptr<FileLocation> location);
        static void readFile(std::shared_ptr<DataFile>file, std::shared_ptr<FileLocation> location, double num_bytes);
        static void writeFile(std::shared_ptr<DataFile>file, std::shared_ptr<FileLocation> location);
        static void createFile(std::shared_ptr<DataFile>file, std::shared_ptr<FileLocation> location);
 
 
        /***********************/
        /***********************/
        bool isScratch();
        void setScratch();
 
        static void copyFile(std::shared_ptr<DataFile>file,
                                     std::shared_ptr<FileLocation> src_location,
                                     std::shared_ptr<FileLocation> dst_location);
 
 
        static void initiateFileCopy(simgrid::s4u::Mailbox *answer_mailbox,
                                      std::shared_ptr<DataFile>file,
                                      std::shared_ptr<FileLocation> src_location,
                                      std::shared_ptr<FileLocation> dst_location);
 
        static void readFiles(std::map<std::shared_ptr<DataFile>, std::shared_ptr<FileLocation>> locations);
 
        static void writeFiles(std::map<std::shared_ptr<DataFile>, std::shared_ptr<FileLocation>> locations);
 
 
        StorageService(const std::string &hostname,
                       const std::set<std::string> mount_points,
                       const std::string &service_name);
 
    protected:
 
        friend class Simulation;
        friend class FileRegistryService;
        friend class FileTransferThread;
 
        static void stageFile(std::shared_ptr<DataFile>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<std::shared_ptr<DataFile>, std::shared_ptr<FileLocation>> locations);
 
        void stageFile(std::shared_ptr<DataFile>file , std::string mountpoint, std::string directory);
 
        bool is_stratch;
 
 
    };
 
    /***********************/
    /***********************/
 
};
 
 
#endif //WRENCH_STORAGESERVICE_H