SimpleStorageService.h

#ifndef WRENCH_SIMPLESTORAGESERVICE_H
#define WRENCH_SIMPLESTORAGESERVICE_H


#include <wrench/services/storage/simple/NetworkConnection.h>
#include <wrench/services/storage/simple/NetworkConnectionManager.h>
#include "wrench/services/storage/StorageService.h"
#include "SimpleStorageServiceProperty.h"
#include "SimpleStorageServiceMessagePayload.h"
#include "wrench/simgrid_S4U_util/S4U_PendingCommunication.h"

namespace wrench {

    class SimulationMessage;
    class SimulationTimestampFileCopyStart;

    class S4U_PendingCommunication;

    class IncomingFile;

    class SimpleStorageService : public StorageService {

    private:
        std::map<std::string, std::string> default_property_values = {
                 {SimpleStorageServiceProperty::MAX_NUM_CONCURRENT_DATA_CONNECTIONS,  "infinity"},
                 {SimpleStorageServiceProperty::SELF_CONNECTION_DELAY,  "0"},
                };

        std::map<std::string, std::string> default_messagepayload_values = {
                 {SimpleStorageServiceMessagePayload::STOP_DAEMON_MESSAGE_PAYLOAD,         "1024"},
                 {SimpleStorageServiceMessagePayload::DAEMON_STOPPED_MESSAGE_PAYLOAD,      "1024"},
                 {SimpleStorageServiceMessagePayload::FREE_SPACE_REQUEST_MESSAGE_PAYLOAD,  "1024"},
                 {SimpleStorageServiceMessagePayload::FREE_SPACE_ANSWER_MESSAGE_PAYLOAD,   "1024"},
                 {SimpleStorageServiceMessagePayload::FILE_DELETE_REQUEST_MESSAGE_PAYLOAD, "1024"},
                 {SimpleStorageServiceMessagePayload::FILE_DELETE_ANSWER_MESSAGE_PAYLOAD,  "1024"},
                 {SimpleStorageServiceMessagePayload::FILE_LOOKUP_REQUEST_MESSAGE_PAYLOAD, "1024"},
                 {SimpleStorageServiceMessagePayload::FILE_LOOKUP_ANSWER_MESSAGE_PAYLOAD,  "1024"},
                 {SimpleStorageServiceMessagePayload::FILE_COPY_REQUEST_MESSAGE_PAYLOAD,   "1024"},
                 {SimpleStorageServiceMessagePayload::FILE_COPY_ANSWER_MESSAGE_PAYLOAD,    "1024"},
                 {SimpleStorageServiceMessagePayload::FILE_WRITE_REQUEST_MESSAGE_PAYLOAD,  "1024"},
                 {SimpleStorageServiceMessagePayload::FILE_WRITE_ANSWER_MESSAGE_PAYLOAD,   "1024"},
                 {SimpleStorageServiceMessagePayload::FILE_READ_REQUEST_MESSAGE_PAYLOAD,   "1024"},
                 {SimpleStorageServiceMessagePayload::FILE_READ_ANSWER_MESSAGE_PAYLOAD,    "1024"},
                };

    public:

        // Public Constructor
        SimpleStorageService(std::string hostname,
                             double capacity,
                             std::map<std::string, std::string> property_list = {},
                             std::map<std::string, std::string> messagepayload_list = {});

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

        ~SimpleStorageService();

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

    private:



        friend class Simulation;

        // Low-level Constructor
        SimpleStorageService(std::string hostname,
                             double capacity,
                             std::map<std::string, std::string> property_list,
                             std::map<std::string, std::string> messagepayload_list,
                             std::string suffix);

        int main() override;

        bool processControlMessage(std::unique_ptr<NetworkConnection> connection);

        bool processDataConnection(std::unique_ptr<NetworkConnection> connection);
        bool processIncomingDataConnection(std::unique_ptr<NetworkConnection> connection);
        bool processOutgoingDataConnection(std::unique_ptr<NetworkConnection> connection);

        unsigned long getNewUniqueNumber();

        bool processFileWriteRequest(WorkflowFile *file, std::string dst_dir, std::string answer_mailbox);

        bool processFileReadRequest(WorkflowFile *file, std::string src_dir, std::string answer_mailbox,
                                    std::string mailbox_to_receive_the_file_content);

        bool processFileCopyRequest(WorkflowFile *file, StorageService *src, std::string src_dir, std::string dst_dir, std::string answer_mailbox, SimulationTimestampFileCopyStart *start_timestamp);

        unsigned long num_concurrent_connections;

        std::unique_ptr<NetworkConnectionManager> network_connection_manager;


    };

};


#endif //WRENCH_SIMPLESTORAGESERVICE_H