blob: 817470deb69215bb73aa29843e768dd142ed2825 (
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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "multienumattribute.hpp"
#include "enummodifier.h"
#include <stdexcept>
namespace search::multienumattribute {
using EnumIndex = IEnumStore::Index;
using EnumIndexRemapper = IEnumStore::EnumIndexRemapper;
using Value = vespalib::datastore::AtomicEntryRef;
using WeightedValue = multivalue::WeightedValue<vespalib::datastore::AtomicEntryRef>;
template <typename WeightedIndex>
void
remap_enum_store_refs(const EnumIndexRemapper& remapper, AttributeVector& v, attribute::MultiValueMapping<WeightedIndex>& multi_value_mapping)
{
// update multi_value_mapping with new EnumIndex values after enum store has been compacted.
v.logEnumStoreEvent("compactfixup", "drain");
{
attribute::EnumModifier enum_guard(v.getEnumModifier());
auto& filter = remapper.get_entry_ref_filter();
v.logEnumStoreEvent("compactfixup", "start");
for (uint32_t doc = 0; doc < v.getNumDocs(); ++doc) {
vespalib::ArrayRef<WeightedIndex> indices(multi_value_mapping.get_writable(doc));
for (auto& entry : indices) {
EnumIndex ref = multivalue::get_value_ref(entry).load_relaxed();
if (ref.valid() && filter.has(ref)) {
ref = remapper.remap(ref);
multivalue::get_value_ref(entry).store_release(ref);
}
}
}
}
v.logEnumStoreEvent("compactfixup", "complete");
}
template void remap_enum_store_refs(const EnumIndexRemapper&, AttributeVector&, attribute::MultiValueMapping<Value> &);
template void remap_enum_store_refs(const EnumIndexRemapper&, AttributeVector&, attribute::MultiValueMapping<WeightedValue> &);
}
|