blob: 682d7a3dc994ce2d3a069bf8d28033787867eec7 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
// 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);
}
};
}
|