searchcore/src/vespa/searchcore/proton/persistenceengine/commit_and_wait_document_retriever.cpp


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

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include "commit_and_wait_document_retriever.h"
#include <vespa/document/fieldvalue/document.h>

namespace proton {

CommitAndWaitDocumentRetriever::CommitAndWaitDocumentRetriever(IDocumentRetriever::SP retriever,
                                                               ILidCommitState & unCommittedLidTracker)
    : _retriever(std::move(retriever)),
      _uncommittedLidsTracker(unCommittedLidTracker)
{ }

CommitAndWaitDocumentRetriever::~CommitAndWaitDocumentRetriever() = default;

const document::DocumentTypeRepo &
CommitAndWaitDocumentRetriever::getDocumentTypeRepo() const {
    return _retriever->getDocumentTypeRepo();
}

void
CommitAndWaitDocumentRetriever::getBucketMetaData(const Bucket &bucket, search::DocumentMetaData::Vector &result) const {
    return _retriever->getBucketMetaData(bucket, result);
}

search::DocumentMetaData
CommitAndWaitDocumentRetriever::getDocumentMetaData(const document::DocumentId &id) const {
    return _retriever->getDocumentMetaData(id);
}

document::Document::UP
CommitAndWaitDocumentRetriever::getFullDocument(search::DocumentIdT lid) const {
    // Ensure that attribute vectors are committed
    _uncommittedLidsTracker.waitComplete(lid);
    return _retriever->getFullDocument(lid);
}

document::Document::UP
CommitAndWaitDocumentRetriever::getPartialDocument(search::DocumentIdT lid, const document::DocumentId & docId,
                                                   const document::FieldSet & fieldSet) const
{
    _uncommittedLidsTracker.waitComplete(lid);
    return _retriever->getPartialDocument(lid, docId, fieldSet);
}

void
CommitAndWaitDocumentRetriever::visitDocuments(const LidVector &lids, search::IDocumentVisitor &visitor,
                                               ReadConsistency readConsistency) const
{
    _uncommittedLidsTracker.waitComplete(lids);
    _retriever->visitDocuments(lids, visitor, readConsistency);
}

CachedSelect::SP
CommitAndWaitDocumentRetriever::parseSelect(const vespalib::string &selection) const {
    return _retriever->parseSelect(selection);
}

IDocumentRetriever::ReadGuard
CommitAndWaitDocumentRetriever::getReadGuard() const {
    return _retriever->getReadGuard();
}

uint32_t
CommitAndWaitDocumentRetriever::getDocIdLimit() const {
    return _retriever->getDocIdLimit();
}

} // namespace proton