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
94
95
96
97
98
99
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "bitvectordictionary.h"
#include <vespa/vespalib/data/fileheader.h>
#include <vespa/fastos/file.h>
#include <cassert>
#include <vespa/log/log.h>
LOG_SETUP(".diskindex.bitvectordictionary");
namespace search::diskindex {
BitVectorDictionary::BitVectorDictionary()
: _docIdLimit(),
_entries(),
_vectorSize(),
_datFile(),
_datHeaderLen(0u)
{ }
BitVectorDictionary::~BitVectorDictionary() = default;
bool
BitVectorDictionary::open(const vespalib::string &pathPrefix,
const TuneFileRandRead &tuneFileRead,
BitVectorKeyScope scope)
{
vespalib::string booloccIdxName = pathPrefix + "boolocc" +
getBitVectorKeyScopeSuffix(scope);
vespalib::string booloccDatName = pathPrefix + "boolocc.bdat";
{
FastOS_File idxFile;
idxFile.OpenReadOnly(booloccIdxName.c_str());
if (!idxFile.IsOpened()) {
LOG(warning, "Could not open bitvector idx file '%s'",
booloccIdxName.c_str());
return false;
}
vespalib::FileHeader idxHeader;
uint32_t idxHeaderLen = idxHeader.readFile(idxFile);
idxFile.SetPosition(idxHeaderLen);
assert(idxHeader.hasTag("frozen"));
assert(idxHeader.hasTag("docIdLimit"));
assert(idxHeader.hasTag("numKeys"));
assert(idxHeader.getTag("frozen").asInteger() != 0);
_docIdLimit = idxHeader.getTag("docIdLimit").asInteger();
uint32_t numEntries = idxHeader.getTag("numKeys").asInteger();
_entries.resize(numEntries);
size_t bufSize = sizeof(WordSingleKey) * numEntries;
assert(idxFile.getSize() >= static_cast<int64_t>(idxHeaderLen + bufSize));
if (bufSize > 0) {
ssize_t has_read = idxFile.Read(&_entries[0], bufSize);
assert(has_read == ssize_t(bufSize));
}
}
_vectorSize = BitVector::getFileBytes(_docIdLimit);
_datFile = std::make_unique<FastOS_File>();
_datFile->setFAdviseOptions(tuneFileRead.getAdvise());
if (tuneFileRead.getWantMemoryMap()) {
_datFile->enableMemoryMap(tuneFileRead.getMemoryMapFlags());
} else if (tuneFileRead.getWantDirectIO()) {
_datFile->EnableDirectIO();
}
_datFile->OpenReadOnly(booloccDatName.c_str());
if (!_datFile->IsOpened()) {
LOG(warning, "Could not open bitvector dat file '%s'",
booloccDatName.c_str());
return false;
}
vespalib::FileHeader datHeader(64);
_datHeaderLen = datHeader.readFile(*_datFile);
assert(_datFile->getSize() >=
static_cast<int64_t>(_vectorSize * _entries.size() + _datHeaderLen));
return true;
}
BitVector::UP
BitVectorDictionary::lookup(uint64_t wordNum)
{
WordSingleKey key;
key._wordNum = wordNum;
auto itr = std::lower_bound(_entries.begin(), _entries.end(), key);
if (itr == _entries.end() || key < *itr) {
return BitVector::UP();
}
int64_t pos = &*itr - &_entries[0];
return BitVector::create(_docIdLimit, *_datFile,
((int64_t) _vectorSize) * pos + _datHeaderLen,
itr->_numDocs);
}
}
|