aboutsummaryrefslogtreecommitdiffstats
path: root/searchlib/src/vespa/searchlib/aggregation/hitsaggregationresult.cpp
blob: 01559952f2864e5e75a6f2bf5b7a3d9e930c0539 (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
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 Yahoo. 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() {}