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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "hitsaggregationresult.h"
#include <vespa/document/fieldvalue/document.h>
#include <vespa/log/log.h>
LOG_SETUP(".searchlib.aggregation.hitsaggregationresult");
namespace search::aggregation {
using vespalib::Serializer;
using vespalib::Deserializer;
IMPLEMENT_IDENTIFIABLE_NS2(search, aggregation, HitsAggregationResult, AggregationResult);
HitsAggregationResult::HitsAggregationResult() :
AggregationResult(),
_summaryClass("default"),
_maxHits(std::numeric_limits<uint32_t>::max()),
_hits(),
_isOrdered(false),
_bestHitRank(),
_summaryGenerator(nullptr)
{}
HitsAggregationResult::HitsAggregationResult(HitsAggregationResult &&) noexcept = default;
HitsAggregationResult & HitsAggregationResult::operator=(HitsAggregationResult &&) noexcept = default;
HitsAggregationResult::HitsAggregationResult(const HitsAggregationResult &) = default;
HitsAggregationResult & HitsAggregationResult::operator=(const HitsAggregationResult &) = default;
HitsAggregationResult::~HitsAggregationResult() = default;
void HitsAggregationResult::onPrepare(const ResultNode & result, bool useForInit)
{
(void) result;
(void) useForInit;
}
void
HitsAggregationResult::onMerge(const AggregationResult &b)
{
const auto &rhs = (const HitsAggregationResult &)b;
_hits.onMerge(rhs._hits);
}
void
HitsAggregationResult::onAggregate(const ResultNode &result, DocId docId, HitRank rank)
{
(void) result;
if ( ! _isOrdered || (_hits.size() < _maxHits)) {
_hits.addHit(FS4Hit(docId, rank), _maxHits);
}
}
void
HitsAggregationResult::onAggregate(const ResultNode & result, const document::Document & doc, HitRank rank)
{
(void) result;
LOG(spam, "Filling vdshit for %s hits=%lu, maxHits=%u", doc.getId().toString().c_str(), (unsigned long)_hits.size(), _maxHits);
if (!_isOrdered || (_hits.size() < _maxHits)) {
VdsHit hit(doc.getId().toString(), rank);
vespalib::ConstBufferRef docsum(_summaryGenerator->fillSummary(0, _summaryClass));
hit.setSummary(docsum.c_str(), docsum.size());
LOG(spam, "actually filled %s with summary %s with blob of size %lu", doc.getId().toString().c_str(),_summaryClass.c_str(), docsum.size() );
_hits.addHit(hit, _maxHits);
}
}
void
HitsAggregationResult::onAggregate(const ResultNode & result)
{
(void) result;
LOG_ABORT("should not reach here");
}
void
HitsAggregationResult::onReset()
{
_hits.clear();
}
Serializer &
HitsAggregationResult::onSerialize(Serializer & os) const
{
AggregationResult::onSerialize(os);
os << _summaryClass << _maxHits;
_hits.serialize(os);
return os;
}
Deserializer &
HitsAggregationResult::onDeserialize(Deserializer & is)
{
AggregationResult::onDeserialize(is);
is >> _summaryClass >> _maxHits;
_hits.deserialize(is);
if (_maxHits == 0) {
_maxHits = std::numeric_limits<uint32_t>::max();
}
return is;
}
void
HitsAggregationResult::visitMembers(vespalib::ObjectVisitor & visitor) const
{
AggregationResult::visitMembers(visitor);
visit(visitor, "summaryClass", _summaryClass);
visit(visitor, "maxHits", _maxHits);
_hits.visitMembers(visitor);
}
void
HitsAggregationResult::selectMembers(const vespalib::ObjectPredicate & predicate, vespalib::ObjectOperation & operation)
{
AggregationResult::selectMembers(predicate, operation);
_hits.selectMembers(predicate, operation);
}
const expression::ResultNode &
HitsAggregationResult::onGetRank() const
{
if ( ! _hits.empty() ) {
_bestHitRank = _hits.front().getRank();
}
return _bestHitRank;
}
}
// this function was added by ../../forcelink.sh
void forcelink_file_searchlib_aggregation_hitsaggregationresult() {}
|