blob: 39c5f40fe13808d4af4501cc3caed54c3ba4cd29 (
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/document/config/documenttypes_config_fwd.h>
#include <vespa/vespalib/stllike/string.h>
#include <functional>
#include <memory>
namespace document {
class ArrayFieldValue;
class DataType;
class Document;
class DocumentType;
class DocumentTypeRepo;
class MapFieldValue;
class StructFieldValue;
class WeightedSetFieldValue;
}
namespace document::config::internal { class InternalDocumenttypesType; }
namespace document::config_builder { struct Struct; }
namespace search::test {
/*
* Class used to make empty search documents.
*/
class DocBuilder {
using DocumenttypesConfig = const document::config::internal::InternalDocumenttypesType;
std::shared_ptr<const DocumenttypesConfig> _document_types_config;
std::shared_ptr<const document::DocumentTypeRepo> _repo;
const document::DocumentType* _document_type;
public:
using AddFieldsType = std::function<void(document::config_builder::Struct&)>;
DocBuilder();
explicit DocBuilder(AddFieldsType add_fields);
~DocBuilder();
const document::DocumentTypeRepo& get_repo() const noexcept { return *_repo; }
std::shared_ptr<const document::DocumentTypeRepo> get_repo_sp() const noexcept { return _repo; }
const document::DocumentType& get_document_type() const noexcept { return *_document_type; }
std::unique_ptr<document::Document> make_document(vespalib::string document_id);
const document::DataType &get_data_type(const vespalib::string &name) const;
const DocumenttypesConfig& get_documenttypes_config() const noexcept { return *_document_types_config; }
document::ArrayFieldValue make_array(vespalib::stringref field_name);
document::WeightedSetFieldValue make_wset(vespalib::stringref field_name);
};
}
|