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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "ucafunctionnode.h"
#include "ucaconverter.h"
namespace search::expression {
using vespalib::Serializer;
using vespalib::Deserializer;
IMPLEMENT_EXPRESSIONNODE(UcaFunctionNode, UnaryFunctionNode);
UcaFunctionNode::UcaFunctionNode() = default;
UcaFunctionNode::~UcaFunctionNode() = default;
UcaFunctionNode::UcaFunctionNode(ExpressionNode::UP arg, const vespalib::string & locale, const vespalib::string & strength) :
UnaryFunctionNode(std::move(arg)),
_locale(locale),
_strength(strength),
_collator(new uca::UcaConverter(locale, strength))
{
}
UcaFunctionNode::UcaFunctionNode(const UcaFunctionNode & rhs) :
UnaryFunctionNode(rhs),
_locale(rhs._locale),
_strength(rhs._strength),
_collator(rhs._collator),
_handler()
{
}
UcaFunctionNode & UcaFunctionNode::operator = (const UcaFunctionNode & rhs)
{
if (this != &rhs) {
UnaryFunctionNode::operator =(rhs);
_locale = rhs._locale;
_strength = rhs._strength;
_collator = rhs._collator;
_handler.reset();
}
return *this;
}
void UcaFunctionNode::onPrepareResult()
{
if (getArg().getResult()->inherits(ResultNodeVector::classId)) {
setResultType(std::make_unique<RawResultNodeVector>());
_handler = std::make_unique<MultiValueHandler>(*this);
} else {
setResultType(std::make_unique<RawResultNode>());
_handler = std::make_unique<SingleValueHandler>(*this);
}
}
UcaFunctionNode::Handler::Handler(const UcaFunctionNode & uca) :
_converter(*uca._collator),
_backingBuffer(),
_buffer(_backingBuffer, sizeof(_backingBuffer))
{
}
void UcaFunctionNode::Handler::handleOne(const ResultNode & arg, RawResultNode & result) const
{
vespalib::ConstBufferRef buf = _converter.convert(arg.getString(_buffer));
result.set(RawResultNode(buf.c_str(), buf.size()));
}
bool UcaFunctionNode::onExecute() const
{
getArg().execute();
_handler->handle(*getArg().getResult());
return true;
}
void UcaFunctionNode::SingleValueHandler::handle(const ResultNode & arg)
{
handleOne(arg, _result);
}
void UcaFunctionNode::MultiValueHandler::handle(const ResultNode & arg)
{
const ResultNodeVector & v(static_cast<const ResultNodeVector &>(arg));
_result.getVector().resize(v.size());
for(size_t i(0), m(_result.getVector().size()); i < m; i++) {
handleOne(v.get(i), _result.getVector()[i]);
}
}
Serializer & UcaFunctionNode::onSerialize(Serializer & os) const
{
UnaryFunctionNode::onSerialize(os);
return os << _locale << _strength;
}
Deserializer & UcaFunctionNode::onDeserialize(Deserializer & is)
{
UnaryFunctionNode::onDeserialize(is);
is >> _locale >> _strength;
_collator.reset(new uca::UcaConverter(_locale, _strength));
return is;
}
}
// this function was added by ../../forcelink.sh
void forcelink_file_searchlib_expression_ucafunctionnode() {}
|