FileRegistryService.h

#ifndef WRENCH_FILEREGISTRYSERVICE_H
#define WRENCH_FILEREGISTRYSERVICE_H

#include <set>
#include <wrench/services/network_proximity/NetworkProximityService.h>

#include "wrench/services/Service.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"},
                };

        std::map<std::string, std::string> 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
        FileRegistryService(std::string hostname,
                            std::map<std::string, std::string> property_list = {},
                            std::map<std::string, std::string> messagepayload_list = {}
        );

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

        std::set<StorageService *> lookupEntry(WorkflowFile *file);

        std::map<double, StorageService *> lookupEntry(WorkflowFile *file, std::string reference_host,
                                                       NetworkProximityService *);

        void addEntry(WorkflowFile *file, StorageService *storage_service);

        void removeEntry(WorkflowFile *file, StorageService *storage_service);

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


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

        ~FileRegistryService();
        /****************************/
        /****************************/

    private:

        friend class Simulation;

        FileRegistryService(std::string hostname,
                            std::map<std::string, std::string> property_list,
                            std::map<std::string, std::string> messagepayload_list,
                            std::string suffix = "");


        void addEntryToDatabase(WorkflowFile *file, StorageService *storage_service);

        bool removeEntryFromDatabase(WorkflowFile *file, StorageService *storage_service);

        int main();

        bool processNextMessage();

        std::map<WorkflowFile *, std::set<StorageService *>> entries;
    };



};


#endif //WRENCH_FILEREGISTRYSERVICE_H