FileRegistryMessage.h

#ifndef WRENCH_FILEREGISTRYMESSAGE_H
#define WRENCH_FILEREGISTRYMESSAGE_H


#include <wrench/services/ServiceMessage.h>
#include <iostream>
#include <wrench/services/network_proximity/NetworkProximityService.h>

namespace wrench {

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

    class FileRegistryMessage : public ServiceMessage {
    protected:
        FileRegistryMessage(std::string name, double payload);

    };

    class FileRegistryFileLookupRequestMessage : public FileRegistryMessage {
    public:
        FileRegistryFileLookupRequestMessage(std::string answer_mailbox, WorkflowFile *file, double payload);

        std::string answer_mailbox;
        WorkflowFile *file;
    };

    class FileRegistryFileLookupAnswerMessage : public FileRegistryMessage {
    public:
        FileRegistryFileLookupAnswerMessage(WorkflowFile *file, std::set<StorageService *> locations, double payload);

        WorkflowFile *file;
        std::set<StorageService *> locations;
    };

    class FileRegistryFileLookupByProximityRequestMessage : public FileRegistryMessage {
    public:
        FileRegistryFileLookupByProximityRequestMessage(std::string answer_mailbox, WorkflowFile *file,
                                                        std::string reference_host,
                NetworkProximityService *network_proximity_service, double payload);

        std::string answer_mailbox;
        WorkflowFile *file;
        std::string reference_host;

        NetworkProximityService *network_proximity_service;
    };

    class FileRegistryFileLookupByProximityAnswerMessage : public FileRegistryMessage {
    public:
        FileRegistryFileLookupByProximityAnswerMessage(WorkflowFile *file,
                                                       std::string reference_host,
                                                       std::map<double, StorageService *> locations,
                                                       double payload);

        WorkflowFile *file;
        std::string reference_host;
        std::map<double, StorageService *> locations;
    };

    class FileRegistryRemoveEntryRequestMessage : public FileRegistryMessage {
    public:
        FileRegistryRemoveEntryRequestMessage(std::string answer_mailbox, WorkflowFile *file,
                                              StorageService *storage_service, double payload);

        std::string answer_mailbox;
        WorkflowFile *file;
        StorageService *storage_service;
    };

    class FileRegistryRemoveEntryAnswerMessage : public FileRegistryMessage {
    public:
        FileRegistryRemoveEntryAnswerMessage(bool success, double payload);

        bool success;
    };

    class FileRegistryAddEntryRequestMessage : public FileRegistryMessage {
    public:
        FileRegistryAddEntryRequestMessage(std::string answer_mailbox, WorkflowFile *file,
                                           StorageService *storage_service, double payload);

        std::string answer_mailbox;
        WorkflowFile *file;
        StorageService *storage_service;
    };

    class FileRegistryAddEntryAnswerMessage : public FileRegistryMessage {
    public:
        FileRegistryAddEntryAnswerMessage(double payload);
    };

    /***********************/
    /***********************/
};


#endif //WRENCH_FILEREGISTRYMESSAGE_H