aboutsummaryrefslogtreecommitdiffstats
path: root/searchcore/src/vespa/searchcore/proton/server/documentretrieverbase.h
blob: fe5a7a1392d12a4f33e7c0dde5a131f9ee2469f7 (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
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#pragma once

#include <vespa/searchcore/proton/persistenceengine/i_document_retriever.h>
#include <vespa/searchcore/proton/common/doctypename.h>
#include <vespa/document/fieldvalue/document.h>
#include <vespa/vespalib/stllike/lrucache_map.h>
#include <vespa/searchlib/attribute/iattributemanager.h>
#include <mutex>

namespace proton {

class DocumentRetrieverBase : public IDocumentRetriever
{
    const DocTypeName                &_docTypeName;
    const document::DocumentTypeRepo &_repo;
    const IDocumentMetaStoreContext  &_meta_store;

    using SelectCache = vespalib::lrucache_map<vespalib::LruParam<vespalib::string, CachedSelect::SP>>;

    mutable SelectCache    _selectCache;
    mutable std::mutex     _lock;
    document::Document::UP _emptyDoc;
    const bool             _hasFields;

protected:
    virtual const search::IAttributeManager * getAttrMgr() const { return nullptr; }
    const document::DocumentType & getDocumentType() const {
        return _emptyDoc->getType();
    }
public:
    DocumentRetrieverBase(const DocTypeName &docTypeName,
                          const document::DocumentTypeRepo &repo,
                          const IDocumentMetaStoreContext &meta_store,
                          bool hasFields);
    ~DocumentRetrieverBase() override;

    const document::DocumentTypeRepo &getDocumentTypeRepo() const override { return _repo; }
    void getBucketMetaData(const storage::spi::Bucket &bucket, search::DocumentMetaData::Vector &result) const override;
    search::DocumentMetaData getDocumentMetaData(const document::DocumentId &id) const override;
    CachedSelect::SP parseSelect(const vespalib::string &selection) const override;
    ReadGuard getReadGuard() const override { return _meta_store.getReadGuard(); }
    uint32_t getDocIdLimit() const override { return _meta_store.getReadGuard()->get().getCommittedDocIdLimit(); }
};

}  // namespace proton