// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once

#include "messagesender.h"
#include <memory>
#include <mutex>
#include <vector>

namespace storage {

class MessageGuard {
    std::vector<std::shared_ptr<api::StorageMessage>> messagesUp;
    std::vector<std::shared_ptr<api::StorageMessage>> messagesDown;

    std::unique_lock<std::mutex> _lock;
    ChainedMessageSender& _messageSender;

public:
    MessageGuard(std::mutex& lock, ChainedMessageSender& messageSender)
        : _lock(lock),
          _messageSender(messageSender)
    {}
    ~MessageGuard();

    void send(const std::shared_ptr<api::StorageMessage>& message) {
        sendUp(message);
    }

    void sendUp(const std::shared_ptr<api::StorageMessage>& message) {
        messagesUp.push_back(message);
    }

    void sendDown(const std::shared_ptr<api::StorageMessage>& message) {
        messagesDown.push_back(message);
    }
};

}