aboutsummaryrefslogtreecommitdiffstats
path: root/vespalib/src/vespa/vespalib/datastore/unique_store_builder.hpp
blob: 9eb61e96ae95b8017d5861bca834285a99d6096d (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
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#pragma once

#include "unique_store_builder.h"
#include "i_unique_store_dictionary.h"
#include "datastore.hpp"

namespace vespalib::datastore {

template <typename Allocator>
UniqueStoreBuilder<Allocator>::UniqueStoreBuilder(Allocator& allocator, IUniqueStoreDictionary& dict, uint32_t uniqueValuesHint)
    : _allocator(allocator),
      _dict(dict),
      _refs(),
      _refCounts()
{
    _refs.reserve(uniqueValuesHint);
    _refs.push_back(EntryRef());
}

template <typename Allocator>
UniqueStoreBuilder<Allocator>::~UniqueStoreBuilder() = default;

template <typename Allocator>
void
UniqueStoreBuilder<Allocator>::setupRefCounts()
{
    _refCounts.resize(_refs.size());
}

template <typename Allocator>
void
UniqueStoreBuilder<Allocator>::makeDictionary()
{
    auto ref_count_itr = _refCounts.cbegin();
    for (auto ref : _refs) {
        auto& wrapped_entry = _allocator.get_wrapped(ref);
        wrapped_entry.set_ref_count(*ref_count_itr);
        ++ref_count_itr;
    }
    _dict.build(_refs, _refCounts, [this](EntryRef ref) { _allocator.hold(ref); });
}

}