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

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

/**
 * \class storage::ServiceLayerManagedComponent
 * \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/document/bucket/bucket.h>

namespace storage {

class ContentBucketSpaceRepo;
class MinimumUsedBitsTracker;
class StorBucketDatabase;

struct ServiceLayerManagedComponent
{
    virtual ~ServiceLayerManagedComponent() = default;

    virtual void setBucketSpaceRepo(ContentBucketSpaceRepo&) = 0;
    virtual void setMinUsedBitsTracker(MinimumUsedBitsTracker&) = 0;
};

struct ServiceLayerComponentRegister : public virtual StorageComponentRegister
{
    virtual void registerServiceLayerComponent(ServiceLayerManagedComponent&) = 0;
};

class ServiceLayerComponent : public StorageComponent,
                              private ServiceLayerManagedComponent
{
    ContentBucketSpaceRepo* _bucketSpaceRepo;
    MinimumUsedBitsTracker* _minUsedBitsTracker;

    // ServiceLayerManagedComponent implementation
    void setBucketSpaceRepo(ContentBucketSpaceRepo& repo) override { _bucketSpaceRepo = &repo; }
    void setMinUsedBitsTracker(MinimumUsedBitsTracker& tracker) override {
        _minUsedBitsTracker = &tracker;
    }
public:
    using UP = std::unique_ptr<ServiceLayerComponent>;

    ServiceLayerComponent(ServiceLayerComponentRegister& compReg,
                          vespalib::stringref name)
        : StorageComponent(compReg, name),
          _bucketSpaceRepo(nullptr),
          _minUsedBitsTracker(nullptr)
    {
        compReg.registerServiceLayerComponent(*this);
    }

    const ContentBucketSpaceRepo &getBucketSpaceRepo() const;
    StorBucketDatabase& getBucketDatabase(document::BucketSpace bucketSpace) const;
    MinimumUsedBitsTracker& getMinUsedBitsTracker() {
        return *_minUsedBitsTracker;
    }
    const MinimumUsedBitsTracker& getMinUsedBitsTracker() const {
        return *_minUsedBitsTracker;
    }
};

} // storage