blob: 6eb3993cef868c1a0c33d48a84115401032ce30a (
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
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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include "indexenvironment.h"
#include <vespa/config-rank-profiles.h>
#include <vespa/searchlib/fef/blueprintfactory.h>
#include <vespa/searchlib/fef/ranksetup.h>
#include <vespa/searchlib/fef/tablemanager.h>
#include <vespa/vsm/vsm/vsm-adapter.h>
namespace streaming {
/**
* This class subscribes to the rank-profiles config and keeps a setup per rank profile.
**/
class RankManager
{
public:
/** collection of field ids for an index **/
using View = std::vector<uint32_t>;
using IRankingAssetsRepo = search::fef::IRankingAssetsRepo;
/**
* This class represents a snapshot of the rank-profiles config with associated setup per rank profile.
* A new instance of this class is created as part of reload config.
**/
class Snapshot {
private:
using NamedPropertySet = std::pair<vespalib::string, search::fef::Properties>;
using ViewMap = vespalib::hash_map<vespalib::string, View>;
using Map = vespalib::hash_map<vespalib::string, int>;
search::fef::TableManager _tableManager;
IndexEnvironment _protoEnv;
std::vector<NamedPropertySet> _properties; // property set per rank profile
std::vector<IndexEnvironment> _indexEnv; // index environment per rank profile
std::vector<std::shared_ptr<const search::fef::RankSetup>> _rankSetup; // rank setup per rank profile
Map _rpmap;
ViewMap _views;
void addProperties(const vespa::config::search::RankProfilesConfig & cfg);
void detectFields(const vsm::VsmfieldsHandle & fields);
void buildFieldMappings(const vsm::VsmfieldsHandle & fields);
bool initRankSetup(const search::fef::BlueprintFactory & factory);
bool setup(const RankManager & manager);
int getIndex(const vespalib::string & key) const {
auto found = _rpmap.find(key);
return (found != _rpmap.end()) ? found->second : 0;
}
public:
Snapshot();
~Snapshot();
const std::vector<NamedPropertySet> & getProperties() const { return _properties; }
bool setup(const RankManager & manager, const vespa::config::search::RankProfilesConfig & cfg, std::shared_ptr<const IRankingAssetsRepo> ranking_assets_repo);
const search::fef::RankSetup & getRankSetup(const vespalib::string &rankProfile) const {
return *(_rankSetup[getIndex(rankProfile)]);
}
const IndexEnvironment & getIndexEnvironment(const vespalib::string &rankProfile) const {
return _indexEnv[getIndex(rankProfile)];
}
const View *getView(const vespalib::string & index) const {
auto itr = _views.find(index);
if (itr != _views.end()) {
return &itr->second;
}
return nullptr;
}
};
private:
search::fef::BlueprintFactory _blueprintFactory;
vespalib::PtrHolder<const Snapshot> _snapshot;
const vsm::VSMAdapter * _vsmAdapter;
void configureRankProfiles(const vespa::config::search::RankProfilesConfig & cfg, std::shared_ptr<const IRankingAssetsRepo> ranking_assets_repo);
virtual void notify(const vsm::VSMConfigSnapshot & snapshot, std::shared_ptr<const IRankingAssetsRepo> ranking_assets_repo);
public:
RankManager(vsm::VSMAdapter * const vsmAdapter);
virtual ~RankManager();
void configure(const vsm::VSMConfigSnapshot & snap, std::shared_ptr<const IRankingAssetsRepo> ranking_assets_repo);
/**
* Retrieves the current snapshot of the rank-profiles config.
**/
std::shared_ptr<const Snapshot> getSnapshot() const { return _snapshot.get(); }
};
} // namespace streaming
|
