blob: f113eee44c6eb552465ca77ba8eda88b103adec7 (
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "sentmessagemap.h"
#include <vespa/storage/distributor/operations/operation.h>
#include <sstream>
#include <set>
#include <cinttypes>
#include <vespa/log/log.h>
LOG_SETUP(".distributor.callback.map");
namespace storage::distributor {
SentMessageMap::SentMessageMap()
: _map()
{
}
SentMessageMap::~SentMessageMap() = default;
Operation*
SentMessageMap::find_by_id_or_nullptr(api::StorageMessage::Id id) const noexcept
{
auto iter = _map.find(id);
return ((iter != _map.end()) ? iter->second.get() : nullptr);
}
std::shared_ptr<Operation>
SentMessageMap::find_by_id_or_empty(api::StorageMessage::Id id) const noexcept
{
auto iter = _map.find(id);
return ((iter != _map.end()) ? iter->second : std::shared_ptr<Operation>());
}
std::shared_ptr<Operation>
SentMessageMap::pop()
{
auto found = _map.begin();
if (found != _map.end()) {
std::shared_ptr<Operation> op = std::move(found->second);
_map.erase(found);
return op;
} else {
return {};
}
}
std::shared_ptr<Operation>
SentMessageMap::pop(api::StorageMessage::Id id)
{
auto found = _map.find(id);
if (found != _map.end()) {
LOG(spam, "Found Id %" PRIu64 " in callback map: %p", id, found->second.get());
std::shared_ptr<Operation> op = std::move(found->second);
_map.erase(found);
return op;
} else {
LOG(spam, "Did not find Id %" PRIu64 " in callback map", id);
return {};
}
}
void
SentMessageMap::insert(api::StorageMessage::Id id, const std::shared_ptr<Operation> & callback)
{
LOG(spam, "Inserting callback %p for message %" PRIu64 "", callback.get(), id);
_map[id] = callback;
}
std::string
SentMessageMap::toString() const
{
std::ostringstream ost;
std::set<std::string> messages;
for (const auto & entry : _map) {
messages.insert(entry.second.get()->toString());
}
for (const auto & message : messages) {
ost << message << "\n";
}
return ost.str();
}
void
SentMessageMap::clear()
{
_map.clear();
}
}
|