blob: dc0573e64603ef51d6aa73ff47ca6b9e6ec6265b (
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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include <vespa/searchcore/proton/common/i_transient_resource_usage_provider.h>
#include <vespa/searchlib/common/indexmetainfo.h>
#include <vespa/searchlib/common/serialnum.h>
#include <vespa/vespalib/stllike/string.h>
#include <vespa/vespalib/util/time.h>
#include <condition_variable>
#include <mutex>
#include <optional>
#include <unordered_map>
namespace proton {
class AttributeDiskLayout;
/*
* Class used to track changes to a directory containing saved
* snapshots of an attribute vector.
*/
class AttributeDirectory
{
public:
class Writer;
using SerialNum = search::SerialNum;
private:
// Keeps track of the disk size (in bytes) for attribute snapshots.
// The disk size is calculated and set when a snapshot is marked as valid.
using SnapshotDiskSizes = std::unordered_map<SerialNum, std::optional<uint64_t>>;
std::weak_ptr<AttributeDiskLayout> _diskLayout;
const vespalib::string _name;
vespalib::system_time _lastFlushTime;
Writer *_writer; // current writer
mutable std::mutex _mutex;
std::condition_variable _cv;
search::IndexMetaInfo _snapInfo;
SnapshotDiskSizes _disk_sizes;
void saveSnapInfo();
vespalib::string getSnapshotDir(SerialNum serialNum) const;
void setLastFlushTime(vespalib::system_time lastFlushTime);
void createInvalidSnapshot(SerialNum serialNum);
void markValidSnapshot(SerialNum serialNum);
void invalidateOldSnapshots(SerialNum serialNum);
void invalidateOldSnapshots();
void removeInvalidSnapshots();
bool removeDiskDir();
void detach();
vespalib::string getDirName() const;
public:
AttributeDirectory(const std::shared_ptr<AttributeDiskLayout> &diskLayout,
const vespalib::string &name);
~AttributeDirectory();
const vespalib::string & getAttrName() const { return _name; }
/*
* Class to make changes to an attribute directory in a
* controlled manner. An exclusive lock is held during lifetime to
* ensure only one active writer at a time for an attribute directory.
*/
class Writer {
AttributeDirectory &_dir;
public:
Writer(AttributeDirectory &dir);
~Writer();
// methods called when saving an attribute.
void setLastFlushTime(vespalib::system_time lastFlushTime) { _dir.setLastFlushTime(lastFlushTime); }
void createInvalidSnapshot(SerialNum serialNum) { _dir.createInvalidSnapshot(serialNum); }
void markValidSnapshot(SerialNum serialNum) { _dir.markValidSnapshot(serialNum); }
vespalib::string getSnapshotDir(SerialNum serialNum) { return _dir.getSnapshotDir(serialNum); }
// methods called while pruning old snapshots or removing attribute
void invalidateOldSnapshots(SerialNum serialNum) { _dir.invalidateOldSnapshots(serialNum); }
void invalidateOldSnapshots() { _dir.invalidateOldSnapshots(); }
void removeInvalidSnapshots() { _dir.removeInvalidSnapshots(); }
bool removeDiskDir() { return _dir.removeDiskDir(); }
void detach() { _dir.detach(); }
};
std::unique_ptr<Writer> getWriter();
std::unique_ptr<Writer> tryGetWriter();
SerialNum getFlushedSerialNum() const;
vespalib::system_time getLastFlushTime() const;
bool empty() const;
vespalib::string getAttributeFileName(SerialNum serialNum);
TransientResourceUsage get_transient_resource_usage() const;
};
} // namespace proton
|