blob: d5104aefea7b70400227b89ca7ba80e8402224b9 (
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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "simplebucketprioritydatabase.h"
#include <vespa/vespalib/stllike/hash_map.hpp>
#include <cassert>
#include <ostream>
#include <sstream>
namespace storage::distributor {
SimpleBucketPriorityDatabase::SimpleBucketPriorityDatabase()
: _pri_fifo_buckets(),
_bucket_to_pri_iterators(),
_fifo_seq_num(0)
{
}
SimpleBucketPriorityDatabase::~SimpleBucketPriorityDatabase() = default;
void
SimpleBucketPriorityDatabase::clearAllEntriesForBucket(const document::Bucket &bucket)
{
auto maybe_iter = _bucket_to_pri_iterators.find(bucket);
if (maybe_iter != _bucket_to_pri_iterators.end()) {
_pri_fifo_buckets.erase(maybe_iter->second);
_bucket_to_pri_iterators.erase(maybe_iter);
}
}
void
SimpleBucketPriorityDatabase::setPriority(const PrioritizedBucket& bucket)
{
clearAllEntriesForBucket(bucket.getBucket());
if (bucket.requiresMaintenance()) {
auto pri_insert_res = _pri_fifo_buckets.emplace(PriFifoCompositeKey(bucket.getPriority(), _fifo_seq_num),
bucket.getBucket());
assert(pri_insert_res.second);
++_fifo_seq_num;
auto inv_insert_res = _bucket_to_pri_iterators.insert(std::make_pair(bucket.getBucket(), pri_insert_res.first));
assert(inv_insert_res.second);
}
}
void
SimpleBucketPriorityDatabase::PriFifoMappingConstIteratorImpl::increment() noexcept
{
if (_pri_fifo_iter != _pri_fifo_end) {
++_pri_fifo_iter;
}
}
bool
SimpleBucketPriorityDatabase::PriFifoMappingConstIteratorImpl::equal(const ConstIteratorImpl& other) const noexcept
{
auto& typed_other = dynamic_cast<const PriFifoMappingConstIteratorImpl&>(other);
return (_pri_fifo_iter == typed_other._pri_fifo_iter);
}
PrioritizedBucket
SimpleBucketPriorityDatabase::PriFifoMappingConstIteratorImpl::dereference() const noexcept
{
assert(_pri_fifo_iter != _pri_fifo_end);
return {_pri_fifo_iter->second, _pri_fifo_iter->first._pri};
}
SimpleBucketPriorityDatabase::const_iterator
SimpleBucketPriorityDatabase::begin() const
{
return const_iterator(std::make_unique<PriFifoMappingConstIteratorImpl>(
_pri_fifo_buckets.begin(), _pri_fifo_buckets.end()));
}
SimpleBucketPriorityDatabase::const_iterator
SimpleBucketPriorityDatabase::end() const
{
return const_iterator(std::make_unique<PriFifoMappingConstIteratorImpl>(
_pri_fifo_buckets.end(), _pri_fifo_buckets.end()));
}
std::string
SimpleBucketPriorityDatabase::toString() const
{
std::ostringstream ss;
for (const auto& e : *this) {
ss << e << '\n';
}
return ss.str();
}
}
|