FileRegistryService.h

 
#ifndef WRENCH_FILEREGISTRYSERVICE_H
#define WRENCH_FILEREGISTRYSERVICE_H
 
#include <set>
 
#include <wrench/services/Service.h>
#include <wrench/services/network_proximity/NetworkProximityService.h>
#include <wrench/services/storage/StorageService.h>
#include <wrench/services/storage/storage_helpers/FileLocation.h>
 
#include "FileRegistryServiceProperty.h"
#include "FileRegistryServiceMessagePayload.h"
 
namespace wrench {
 
    class WorkflowFile;
 
    class StorageService;
 
    class FileRegistryService : public Service {
 
    public:
 
 
    private:
 
        std::map<std::string, std::string> default_property_values = {
                {FileRegistryServiceProperty::LOOKUP_COMPUTE_COST,                      "0.0"},
                {FileRegistryServiceProperty::ADD_ENTRY_COMPUTE_COST,                   "0.0"},
                {FileRegistryServiceProperty::REMOVE_ENTRY_COMPUTE_COST,                "0.0"},
        };
 
        std::map<std::string, double> default_messagepayload_values = {
                {FileRegistryServiceMessagePayload::STOP_DAEMON_MESSAGE_PAYLOAD,          1024},
                {FileRegistryServiceMessagePayload::DAEMON_STOPPED_MESSAGE_PAYLOAD,       1024},
                {FileRegistryServiceMessagePayload::FILE_LOOKUP_REQUEST_MESSAGE_PAYLOAD,  1024},
                {FileRegistryServiceMessagePayload::FILE_LOOKUP_ANSWER_MESSAGE_PAYLOAD,   1024},
                {FileRegistryServiceMessagePayload::REMOVE_ENTRY_REQUEST_MESSAGE_PAYLOAD, 1024},
                {FileRegistryServiceMessagePayload::REMOVE_ENTRY_ANSWER_MESSAGE_PAYLOAD,  1024},
                {FileRegistryServiceMessagePayload::ADD_ENTRY_REQUEST_MESSAGE_PAYLOAD,    1024},
                {FileRegistryServiceMessagePayload::ADD_ENTRY_ANSWER_MESSAGE_PAYLOAD,     1024},
        };
 
    public:
 
 
        // Public Constructor
        explicit FileRegistryService(std::string hostname,
                                     std::map<std::string, std::string> property_list = {},
                                     std::map<std::string, double> messagepayload_list = {}
        );
 
        /****************************/
        /****************************/
 
        std::set<std::shared_ptr<FileLocation>> lookupEntry(WorkflowFile *file);
 
        std::map<double, std::shared_ptr<FileLocation>> lookupEntry(WorkflowFile *file, std::string reference_host,
                                                                    std::shared_ptr<NetworkProximityService> network_proximity_service);
 
        void addEntry(WorkflowFile *file, std::shared_ptr<FileLocation> location);
 
        void removeEntry(WorkflowFile *file, std::shared_ptr<FileLocation> location);
 
        /****************************/
        /****************************/
 
 
        /****************************/
        /****************************/
 
        ~FileRegistryService() override;
        /****************************/
        /****************************/
 
    private:
 
        friend class Simulation;
 
        void addEntryToDatabase(WorkflowFile *file, std::shared_ptr<FileLocation> location);
 
        bool removeEntryFromDatabase(WorkflowFile *file, std::shared_ptr<FileLocation> location);
 
        int main() override;
 
        bool processNextMessage();
 
        std::map<WorkflowFile *, std::set<std::shared_ptr<FileLocation>>> entries;
    };
 
 
 
};
 
 
#endif //WRENCH_FILEREGISTRYSERVICE_H