blob: e8d7320a03644a2be4b4b6cfe30db687753d2b97 (
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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include "multinumericattribute.h"
#include "multi_numeric_search_context.h"
#include <vespa/searchlib/common/growablebitvector.h>
#include <vespa/vespalib/datastore/atomic_value_wrapper.h>
namespace search {
using FlagBaseImpl = MultiValueNumericAttribute< IntegerAttributeTemplate<int8_t>, int8_t >;
template <typename B>
class FlagAttributeT : public B {
public:
FlagAttributeT(const vespalib::string & baseFileName, const AttributeVector::Config & cfg);
private:
using DocId = AttributeVector::DocId;
bool onLoad(vespalib::Executor *executor) override;
bool onLoadEnumerated(ReaderBase &attrReader) override;
std::unique_ptr<attribute::SearchContext>
getSearch(std::unique_ptr<QueryTermSimple> term, const attribute::SearchContextParams & params) const override;
void clearOldValues(DocId doc) override;
void setNewValues(DocId doc, const std::vector<typename B::WType> & values) override;
public:
void setNewBVValue(DocId doc, multivalue::ValueType_t<typename B::WType> value);
private:
bool onAddDoc(DocId doc) override;
void onAddDocs(DocId docIdLimit) override;
void ensureGuardBit(BitVector & bv);
void ensureGuardBit();
void clearGuardBit(DocId doc);
void resizeBitVectors(uint32_t neededSize);
void reclaim_memory(vespalib::GenerationHandler::generation_t oldest_used_gen) override;
uint32_t getOffset(int8_t value) const { return value + 128; }
using AtomicBitVectorPtr = vespalib::datastore::AtomicValueWrapper<BitVector *>;
vespalib::GenerationHolder _bitVectorHolder;
std::vector<std::shared_ptr<GrowableBitVector> > _bitVectorStore;
std::vector<AtomicBitVectorPtr> _bitVectors;
uint32_t _bitVectorSize;
};
using FlagAttribute = FlagAttributeT<FlagBaseImpl>;
} // namespace search
|