aboutsummaryrefslogtreecommitdiffstats
path: root/slobrok/src/vespa/slobrok/server/sbenv.h
blob: cdfe5a21667b5419f8a8a26541e0bcb047c1b9a7 (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
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once

#include "named_service.h"
#include "rpc_mapping_monitor.h"
#include "remote_slobrok.h"
#include "exchange_manager.h"
#include "configshim.h"
#include "ok_state.h"
#include "local_rpc_monitor_map.h"
#include "metrics_producer.h"
#include "union_service_map.h"
#include <vespa/config-slobroks.h>
#include <vespa/slobrok/cfg.h>
#include <vespa/vespalib/net/http/simple_health_producer.h>
#include <vespa/vespalib/net/http/simple_component_config_producer.h>

class FNET_Transport;
class FNET_Scheduler;
class FRT_Supervisor;

namespace slobrok {

class NamedService;
class ManagedRpcServer;
class RPCHooks;
class RemoteCheck;

/**
 * @class SBEnv
 * @brief Environmental class containing an entire server location broker
 *
 * XXX more description needed
 **/
class SBEnv : public Configurable
{
private:
    std::unique_ptr<FNET_Transport>    _transport;
    std::unique_ptr<FRT_Supervisor>    _supervisor;

    ConfigShim                     _configShim;
    std::unique_ptr<Configurator>  _configurator;
    bool                           _shuttingDown;

    SBEnv(const SBEnv &);            // Not used
    SBEnv &operator=(const SBEnv &); // Not used

    void setup(const std::vector<std::string> &cfg) override;

    std::vector<std::string>                   _partnerList;
    std::string                                _me;
    LocalRpcMonitorMap                         _localRpcMonitorMap;
    ServiceMapHistory                          _globalVisibleHistory;
    RPCHooks                                   _rpcHooks;
    std::unique_ptr<RemoteCheck>               _remotechecktask;
    vespalib::SimpleHealthProducer             _health;
    MetricsProducer                            _metrics;
    vespalib::SimpleComponentConfigProducer    _components;
    UnionServiceMap                            _consensusMap;

    ExchangeManager                            _exchanger;

    std::unique_ptr<MapSubscription>           _localMonitorSubscription;
    std::unique_ptr<MapSubscription>           _consensusSubscription;
    std::unique_ptr<MapSubscription>           _globalHistorySubscription;

public:
    explicit SBEnv(const ConfigShim &shim);
    ~SBEnv();

    FNET_Transport *getTransport() { return _transport.get(); }
    FNET_Scheduler *getScheduler();
    FRT_Supervisor *getSupervisor() { return _supervisor.get(); }

    void shutdown();
    void suspend();
    void resume();

    ExchangeManager& exchangeManager() { return _exchanger; }

    ServiceMapHistory& globalHistory() {
        return _globalVisibleHistory;
    }

    LocalRpcMonitorMap& localMonitorMap() {
        return _localRpcMonitorMap;
    }

    ServiceMapHistory& localHistory() {
        return _localRpcMonitorMap.history();
    }

    UnionServiceMap& consensusMap() {
        return _consensusMap;
    }

    const std::string & mySpec() const { return _me; }

    bool isSuspended() const { return false; }
    bool isShuttingDown() const { return _shuttingDown; }

    int MainLoop();

    OkState addPeer(const std::string& name, const std::string& spec);
    OkState removePeer(const std::string& name, const std::string& spec);

    void countFailedHeartbeat() {
        _rpcHooks.countFailedHeartbeat();
    }
    void setConsensusTime(unsigned long value) {
        _rpcHooks.setConsensusTime(value);
    }
};

} // namespace slobrok