SimpleStorageService.h

 
#ifndef WRENCH_SIMPLESTORAGESERVICE_H
#define WRENCH_SIMPLESTORAGESERVICE_H
 
 
#include <wrench/services/storage/storage_helpers/FileTransferThread.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 SimpleStorageService : public StorageService {
 
    private:
        std::map<std::string, std::string> default_property_values = {
                 {SimpleStorageServiceProperty::MAX_NUM_CONCURRENT_DATA_CONNECTIONS,  "infinity"},
                 {SimpleStorageServiceProperty::BUFFER_SIZE,  "1048576"}, // 1 MEGA BYTE
                };
 
        std::map<std::string, double> 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,
                             std::set<std::string> mount_points,
                             std::map<std::string, std::string> property_list = {},
                             std::map<std::string, double> messagepayload_list = {});
 
        /***********************/
        /***********************/
 
        ~SimpleStorageService() override;
        void cleanup(bool has_returned_from_main, int return_value) override;
 
 
        /***********************/
        /***********************/
 
    private:
 
        friend class Simulation;
 
        // Low-level Constructor
        SimpleStorageService(std::string hostname,
                             std::set<std::string> mount_points,
                             std::map<std::string, std::string> property_list,
                             std::map<std::string, double> messagepayload_list,
                             std::string suffix);
 
        int main() override;
 
        bool processNextMessage();
 
        unsigned long getNewUniqueNumber();
 
        bool processFileDeleteRequest(WorkflowFile *file, std::shared_ptr<FileLocation> location, std::string answer_mailbox);
 
        bool processFileWriteRequest(WorkflowFile *file, std::shared_ptr<FileLocation>, std::string answer_mailbox, unsigned long buffer_size);
 
        bool processFileReadRequest(WorkflowFile *file, std::shared_ptr<FileLocation> location, std::string answer_mailbox,
                                    std::string mailbox_to_receive_the_file_content, unsigned long buffer_size);
 
        bool processFileCopyRequest(WorkflowFile *file,
                std::shared_ptr<FileLocation> src,
                std::shared_ptr<FileLocation> dst,
                std::string answer_mailbox);
 
        bool processFileTransferThreadNotification(
                std::shared_ptr<FileTransferThread> ftt,
                WorkflowFile *file,
                std::string src_mailbox,
                std::shared_ptr<FileLocation> src_location,
                std::string dst_mailbox,
                std::shared_ptr<FileLocation> dst_location,
                bool success,
                std::shared_ptr<FailureCause> failure_cause,
                std::string answer_mailbox_if_read,
                std::string answer_mailbox_if_write,
                std::string answer_mailbox_if_copy);
 
        unsigned long num_concurrent_connections;
 
        void startPendingFileTransferThread();
 
        std::deque<std::shared_ptr<FileTransferThread>> pending_file_transfer_threads;
        std::set<std::shared_ptr<FileTransferThread>> running_file_transfer_threads;
 
        void validateProperties();
 
    };
 
};
 
 
#endif //WRENCH_SIMPLESTORAGESERVICE_H