FileRegistryMessage.h

 
#ifndef WRENCH_FILEREGISTRYMESSAGE_H
#define WRENCH_FILEREGISTRYMESSAGE_H
 
 
#include <iostream>
#include <wrench/services/ServiceMessage.h>
#include <wrench/workflow/WorkflowFile.h>
#include <wrench/services/network_proximity/NetworkProximityService.h>
#include <wrench/services/storage/StorageService.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<std::shared_ptr<FileLocation>> locations,
                                            double payload);
 
        WorkflowFile *file;
        std::set<std::shared_ptr<FileLocation>> locations;
    };
 
    class FileRegistryFileLookupByProximityRequestMessage : public FileRegistryMessage {
    public:
        FileRegistryFileLookupByProximityRequestMessage(std::string answer_mailbox, WorkflowFile *file,
                                                        std::string reference_host,
                                                        std::shared_ptr<NetworkProximityService> network_proximity_service,
                                                        double payload);
 
        std::string answer_mailbox;
        WorkflowFile *file;
        std::string reference_host;
 
        std::shared_ptr<NetworkProximityService> network_proximity_service;
    };
 
    class FileRegistryFileLookupByProximityAnswerMessage : public FileRegistryMessage {
    public:
        FileRegistryFileLookupByProximityAnswerMessage(WorkflowFile *file,
                                                       std::string reference_host,
                                                       std::map<double, std::shared_ptr<FileLocation>> locations,
                                                       double payload);
 
        WorkflowFile *file;
        std::string reference_host;
        std::map<double, std::shared_ptr<FileLocation>> locations;
    };
 
    class FileRegistryRemoveEntryRequestMessage : public FileRegistryMessage {
    public:
        FileRegistryRemoveEntryRequestMessage(std::string answer_mailbox, WorkflowFile *file,
                                              std::shared_ptr<FileLocation> location, double payload);
 
        std::string answer_mailbox;
        WorkflowFile *file;
        std::shared_ptr<FileLocation> location;
    };
 
    class FileRegistryRemoveEntryAnswerMessage : public FileRegistryMessage {
    public:
        FileRegistryRemoveEntryAnswerMessage(bool success, double payload);
 
        bool success;
    };
 
    class FileRegistryAddEntryRequestMessage : public FileRegistryMessage {
    public:
        FileRegistryAddEntryRequestMessage(std::string answer_mailbox, WorkflowFile *file,
                                           std::shared_ptr<FileLocation> location, double payload);
 
        std::string answer_mailbox;
        WorkflowFile *file;
        std::shared_ptr<FileLocation> location;
    };
 
    class FileRegistryAddEntryAnswerMessage : public FileRegistryMessage {
    public:
        FileRegistryAddEntryAnswerMessage(double payload);
    };
 
    /***********************/
    /***********************/
};
 
 
#endif //WRENCH_FILEREGISTRYMESSAGE_H