LogicalFileSystem.h

 
#ifndef WRENCH_LOGICALFILESYSTEM_H
#define WRENCH_LOGICALFILESYSTEM_H
 
#include <stdexcept>
#include <string>
#include <map>
#include <set>
 
 
#include "wrench/workflow/WorkflowFile.h"
 
namespace wrench {
 
    /***********************/
    /***********************/
 
 
    class StorageService;
 
    class LogicalFileSystem {
 
    public:
 
        explicit LogicalFileSystem(std::string hostname, StorageService *storage_service, std::string mount_point);
 
        void init();
 
        double getTotalCapacity();
        bool hasEnoughFreeSpace(double bytes);
        double getFreeSpace();
        void reserveSpace(WorkflowFile *file, std::string absolute_path);
        void unreserveSpace(WorkflowFile *file, std::string absolute_path);
 
        void createDirectory(std::string absolute_path);
        bool doesDirectoryExist(std::string absolute_path);
        bool isDirectoryEmpty(std::string absolute_path);
        void removeEmptyDirectory(std::string absolute_path);
        void storeFileInDirectory(WorkflowFile *file, std::string absolute_path);
        void removeFileFromDirectory(WorkflowFile *file, std::string absolute_path);
        void removeAllFilesInDirectory(std::string absolute_path);
        bool isFileInDirectory(WorkflowFile *file, std::string absolute_path);
        std::set<WorkflowFile *> listFilesInDirectory(std::string absolute_path);
 
 
    private:
 
        friend class StorageService;
 
        void stageFile(WorkflowFile *file, std::string absolute_path);
 
        static std::map<std::string, StorageService*> mount_points;
 
        std::map<std::string, std::set<WorkflowFile*>> content;
 
        std::string hostname;
        StorageService *storage_service;
        std::string mount_point;
        double total_capacity;
        double occupied_space;
        std::map<std::string, double> reserved_space;
 
        bool initialized;
 
        void assertInitHasBeenCalled() {
            if (not this->initialized) {
                throw std::runtime_error("LogicalFileSystem::assertInitHasBeenCalled(): A logical file system needs to be initialized before it's been called");
            }
        }
        void assertDirectoryExist(std::string absolute_path) {
            if (not this->doesDirectoryExist(absolute_path)) {
                throw std::invalid_argument("LogicalFileSystem::assertDirectoryExists(): directory " + absolute_path + " does not exist");
            }
        }
 
        void assertDirectoryDoesNotExist(std::string absolute_path) {
            if (this->doesDirectoryExist(absolute_path)) {
                throw std::invalid_argument("LogicalFileSystem::assertDirectoryExists(): directory " + absolute_path + " already exists");
            }
        }
 
        void assertDirectoryIsEmpty(std::string absolute_path) {
            assertDirectoryExist(absolute_path);
            if (not this->isDirectoryEmpty(absolute_path)) {
                throw std::invalid_argument("LogicalFileSystem::assertDirectoryIsEmpty(): directory " + absolute_path + "is not empty");
            }
        }
 
        void assertFileIsInDirectory(WorkflowFile *file, std::string absolute_path) {
            assertDirectoryExist(absolute_path);
            if (this->content[absolute_path].find(file) == this->content[absolute_path].end()) {
                throw std::invalid_argument("LogicalFileSystem::assertFileIsInDirectory(): File " + file->getID() +
                " is not in directory " + absolute_path);
            }
        }
 
    };
 
 
    /***********************/
    /***********************/
 
}
 
 
#endif //WRENCH_LOGICALFILESYSTEM_H