aboutsummaryrefslogtreecommitdiffstats
path: root/searchlib/src/vespa/searchlib/attribute/predicate_attribute_saver.h
blob: b0fc835a34b003d401464472cee03d8deece684c (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
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#pragma once

#include "attributesaver.h"
#include <vespa/vespalib/stllike/allocator.h>
#include <memory>

namespace search::predicate { class ISaver; }

namespace search {

/**
 * Class for saving a predicate attribute.
 */
class PredicateAttributeSaver : public AttributeSaver
{
public:
    using GenerationHandler = vespalib::GenerationHandler;
    using MinFeatureVector = std::vector<uint8_t, vespalib::allocator_large<uint8_t>>;
    using IntervalRangeVector = std::vector<uint16_t, vespalib::allocator_large<uint16_t>>;

private:
    uint32_t                           _version;
    std::unique_ptr<predicate::ISaver> _index_saver;
    MinFeatureVector                   _min_feature;
    IntervalRangeVector                _interval_range_vector;
    uint16_t                           _max_interval_range;
public:
    PredicateAttributeSaver(GenerationHandler::Guard&& guard,
                            const attribute::AttributeHeader& header,
                            uint32_t version,
                            std::unique_ptr<predicate::ISaver> index_saver,
                            MinFeatureVector min_feature,
                            IntervalRangeVector interval_range_vector,
                            uint16_t max_interval_range);
    ~PredicateAttributeSaver() override;
    bool onSave(IAttributeSaveTarget& save_target) override;
};

}