aboutsummaryrefslogtreecommitdiffstats
path: root/storage/src/vespa/storage/common/distributorcomponent.h
blob: 1892c7ce5a2864db2ff99371fc70f1d3481208c8 (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
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
/**
 * \class storage::DistributorStripeComponent
 * \ingroup common
 *
 * \brief Component class including some service layer specific information.
 */

/**
 * \class storage::DistributorComponentRegister
 * \ingroup common
 *
 * \brief Specialization of ComponentRegister handling service layer components.
 */

/**
 * \class storage::DistributorManagedComponent
 * \ingroup common
 *
 * \brief Specialization of StorageManagedComponent.
 *
 * A service layer component register will use this interface in order to set
 * the service layer functionality parts.
 */

#pragma once

#include "storagecomponent.h"
#include <vespa/storage/bucketdb/bucketdatabase.h>
#include <vespa/storage/config/distributorconfiguration.h>
#include <vespa/storage/config/config-stor-distributormanager.h>
#include <vespa/storage/config/config-stor-visitordispatcher.h>
#include <vespa/storageapi/defs.h>

namespace storage {

using DistributorConfig = vespa::config::content::core::internal::InternalStorDistributormanagerType;
using VisitorConfig = vespa::config::content::core::internal::InternalStorVisitordispatcherType;

struct UniqueTimeCalculator {
    virtual ~UniqueTimeCalculator() = default;
    [[nodiscard]] virtual api::Timestamp generate_unique_timestamp() = 0;
};

struct DistributorManagedComponent
{
    virtual ~DistributorManagedComponent() = default;

    virtual void setTimeCalculator(UniqueTimeCalculator&) = 0;
    virtual void setDistributorConfig(const DistributorConfig&)= 0;
    virtual void setVisitorConfig(const VisitorConfig&) = 0;
};

struct DistributorComponentRegister : public virtual StorageComponentRegister
{
    virtual void registerDistributorComponent(DistributorManagedComponent&) = 0;
};

class DistributorComponent : public StorageComponent,
                             private DistributorManagedComponent
{
    mutable UniqueTimeCalculator* _timeCalculator;
    DistributorConfig             _distributorConfig;
    VisitorConfig                 _visitorConfig;
    uint64_t                      _internal_config_generation; // Note: NOT related to config system generations
    std::shared_ptr<const DistributorConfiguration> _config_snapshot;

    void setTimeCalculator(UniqueTimeCalculator& utc) override { _timeCalculator = &utc; }
    void setDistributorConfig(const DistributorConfig& c) override {
        _distributorConfig = c;
        update_config_snapshot();
    }
    void setVisitorConfig(const VisitorConfig& c) override {
        _visitorConfig = c;
        update_config_snapshot();
    }

    void update_config_snapshot();

public:
    using UP = std::unique_ptr<DistributorComponent>;

    DistributorComponent(DistributorComponentRegister& compReg, vespalib::stringref name);
    ~DistributorComponent() override;

    [[nodiscard]] api::Timestamp getUniqueTimestamp() const {
        return _timeCalculator->generate_unique_timestamp();
    }
    const DistributorConfig& getDistributorConfig() const {
        return _distributorConfig;
    }
    const VisitorConfig& getVisitorConfig() const {
        return _visitorConfig;
    }
    uint64_t internal_config_generation() const noexcept {
        return _internal_config_generation;
    }
    std::shared_ptr<const DistributorConfiguration> total_distributor_config_sp() const noexcept {
        return _config_snapshot;
    }
};

} // storage