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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "intrinsic_blueprint_adapter.h"
#include <vespa/searchlib/fef/fef.h>
using search::fef::Blueprint;
using search::fef::FeatureExecutor;
using search::fef::FeatureNameBuilder;
using search::fef::FeatureType;
namespace search::features::rankingexpression {
namespace {
bool is_valid(const std::optional<FeatureType> &type) {
if (!type.has_value()) {
return false;
}
if (!type.value().is_object()) {
return true;
}
return !type.value().type().is_error();
}
struct IntrinsicBlueprint : IntrinsicExpression {
Blueprint::UP blueprint;
FeatureType type;
IntrinsicBlueprint(Blueprint::UP blueprint_in, const FeatureType &type_in)
: blueprint(std::move(blueprint_in)), type(type_in) {}
vespalib::string describe_self() const override { return blueprint->getName(); }
const FeatureType &result_type() const override { return type; }
void prepare_shared_state(const QueryEnv & env, fef::IObjectStore & store) const override {
blueprint->prepareSharedState(env, store);
}
FeatureExecutor &create_executor(const QueryEnv &env, vespalib::Stash &stash) const override {
return blueprint->createExecutor(env, stash);
}
};
struct ResultTypeExtractor : Blueprint::DependencyHandler {
std::optional<FeatureType> result_type;
bool too_much;
bool failed;
ResultTypeExtractor() : result_type(), too_much(false), failed(false) {}
std::optional<FeatureType> resolve_input(const vespalib::string &, Blueprint::AcceptInput) override {
too_much = true;
return std::nullopt;
}
void define_output(const vespalib::string &, FeatureType type) override {
too_much = (too_much || result_type.has_value());
result_type.emplace(std::move(type));
}
void fail(const vespalib::string &) override { failed = true; }
bool valid() const { return (is_valid(result_type) && !too_much && !failed); }
const FeatureType &get() const { return result_type.value(); }
};
} // namespace search::features::rankingexpression::<unnamed>
IntrinsicExpression::UP
IntrinsicBlueprintAdapter::try_create(const search::fef::Blueprint &proto,
const search::fef::IIndexEnvironment &env,
const std::vector<vespalib::string> ¶ms)
{
FeatureNameBuilder name_builder;
ResultTypeExtractor result_type;
Blueprint::UP blueprint = proto.createInstance();
name_builder.baseName(blueprint->getBaseName());
for (const auto ¶m: params) {
name_builder.parameter(param);
}
blueprint->setName(name_builder.buildName());
blueprint->attach_dependency_handler(result_type);
if (!blueprint->setup(env, params) || !result_type.valid()) {
return IntrinsicExpression::UP(nullptr);
}
blueprint->detach_dependency_handler();
return std::make_unique<IntrinsicBlueprint>(std::move(blueprint), result_type.get());
}
} // namespace search::features::rankingexpression
|