FileRegistryMessage.h

 
#ifndef WRENCH_FILEREGISTRYMESSAGE_H
#define WRENCH_FILEREGISTRYMESSAGE_H
 
 
#include <iostream>
#include "wrench/services/ServiceMessage.h"
#include "wrench/data_file/DataFile.h"
#include "wrench/services/network_proximity/NetworkProximityService.h"
#include "wrench/services/storage/StorageService.h"
 
namespace wrench {
 
    /***********************/
    /***********************/
 
    class FileRegistryMessage : public ServiceMessage {
    protected:
        FileRegistryMessage(double payload);
 
    };
 
    class FileRegistryFileLookupRequestMessage : public FileRegistryMessage {
    public:
        FileRegistryFileLookupRequestMessage(simgrid::s4u::Mailbox *answer_mailbox, std::shared_ptr<DataFile>file, double payload);
 
        simgrid::s4u::Mailbox *answer_mailbox;
        std::shared_ptr<DataFile>file;
    };
 
    class FileRegistryFileLookupAnswerMessage : public FileRegistryMessage {
    public:
        FileRegistryFileLookupAnswerMessage(std::shared_ptr<DataFile>file, std::set<std::shared_ptr<FileLocation>> locations,
                                            double payload);
 
        std::shared_ptr<DataFile>file;
        std::set<std::shared_ptr<FileLocation>> locations;
    };
 
    class FileRegistryFileLookupByProximityRequestMessage : public FileRegistryMessage {
    public:
        FileRegistryFileLookupByProximityRequestMessage(simgrid::s4u::Mailbox *answer_mailbox, std::shared_ptr<DataFile>file,
                                                        std::string reference_host,
                                                        std::shared_ptr<NetworkProximityService> network_proximity_service,
                                                        double payload);
 
        simgrid::s4u::Mailbox *answer_mailbox;
        std::shared_ptr<DataFile>file;
        std::string reference_host;
 
        std::shared_ptr<NetworkProximityService> network_proximity_service;
    };
 
    class FileRegistryFileLookupByProximityAnswerMessage : public FileRegistryMessage {
    public:
        FileRegistryFileLookupByProximityAnswerMessage(std::shared_ptr<DataFile>file,
                                                       std::string reference_host,
                                                       std::map<double, std::shared_ptr<FileLocation>> locations,
                                                       double payload);
 
        std::shared_ptr<DataFile>file;
        std::string reference_host;
        std::map<double, std::shared_ptr<FileLocation>> locations;
    };
 
    class FileRegistryRemoveEntryRequestMessage : public FileRegistryMessage {
    public:
        FileRegistryRemoveEntryRequestMessage(simgrid::s4u::Mailbox *answer_mailbox, std::shared_ptr<DataFile>file,
                                              std::shared_ptr<FileLocation> location, double payload);
 
        simgrid::s4u::Mailbox *answer_mailbox;
        std::shared_ptr<DataFile>file;
        std::shared_ptr<FileLocation> location;
    };
 
    class FileRegistryRemoveEntryAnswerMessage : public FileRegistryMessage {
    public:
        FileRegistryRemoveEntryAnswerMessage(bool success, double payload);
 
        bool success;
    };
 
    class FileRegistryAddEntryRequestMessage : public FileRegistryMessage {
    public:
        FileRegistryAddEntryRequestMessage(simgrid::s4u::Mailbox *answer_mailbox, std::shared_ptr<DataFile>file,
                                           std::shared_ptr<FileLocation> location, double payload);
 
        simgrid::s4u::Mailbox *answer_mailbox;
        std::shared_ptr<DataFile>file;
        std::shared_ptr<FileLocation> location;
    };
 
    class FileRegistryAddEntryAnswerMessage : public FileRegistryMessage {
    public:
        FileRegistryAddEntryAnswerMessage(double payload);
    };
 
    /***********************/
    /***********************/
};
 
 
#endif //WRENCH_FILEREGISTRYMESSAGE_H