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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
|
// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
#include "cacheshared.h"
#include <zencore/stats.h>
#include <zenstore/blockstore.h>
#include <zenstore/caslog.h>
ZEN_THIRD_PARTY_INCLUDES_START
#include <tsl/robin_map.h>
ZEN_THIRD_PARTY_INCLUDES_END
#include <filesystem>
namespace zen {
class IoBuffer;
#pragma pack(push)
#pragma pack(1)
struct DiskLocation
{
inline DiskLocation() = default;
inline DiskLocation(uint64_t ValueSize, uint8_t Flags) : Flags(Flags | kStandaloneFile) { Location.StandaloneSize = ValueSize; }
inline DiskLocation(const BlockStoreLocation& Location, uint64_t PayloadAlignment, uint8_t Flags) : Flags(Flags & ~kStandaloneFile)
{
this->Location.BlockLocation = BlockStoreDiskLocation(Location, PayloadAlignment);
}
inline bool operator!=(const DiskLocation& Rhs) const { return memcmp(&Location, &Rhs.Location, sizeof(Location)) != 0; }
inline BlockStoreLocation GetBlockLocation(uint64_t PayloadAlignment) const
{
ZEN_ASSERT(!(Flags & kStandaloneFile));
return Location.BlockLocation.Get(PayloadAlignment);
}
inline uint64_t Size() const { return (Flags & kStandaloneFile) ? Location.StandaloneSize : Location.BlockLocation.GetSize(); }
inline uint8_t IsFlagSet(uint64_t Flag) const { return Flags & Flag; }
inline uint8_t GetFlags() const { return Flags; }
inline ZenContentType GetContentType() const
{
ZenContentType ContentType = ZenContentType::kBinary;
if (IsFlagSet(kStructured))
{
ContentType = ZenContentType::kCbObject;
}
if (IsFlagSet(kCompressed))
{
ContentType = ZenContentType::kCompressedBinary;
}
return ContentType;
}
union
{
BlockStoreDiskLocation BlockLocation; // 10 bytes
uint64_t StandaloneSize = 0; // 8 bytes
} Location;
static const uint8_t kStandaloneFile = 0x80u; // Stored as a separate file
static const uint8_t kStructured = 0x40u; // Serialized as compact binary
static const uint8_t kTombStone = 0x20u; // Represents a deleted key/value
static const uint8_t kCompressed = 0x10u; // Stored in compressed buffer format
uint8_t Flags = 0;
uint8_t Reserved = 0;
};
struct DiskIndexEntry
{
IoHash Key; // 20 bytes
DiskLocation Location; // 12 bytes
};
#pragma pack(pop)
static_assert(sizeof(DiskIndexEntry) == 32);
//////////////////////////////////////////////////////////////////////////
class ZenCacheDiskLayer
{
public:
struct Configuration
{
std::filesystem::path RootDir;
bool EnableReferenceCaching;
};
struct BucketInfo
{
uint64_t EntryCount = 0;
uint64_t TotalSize = 0;
};
struct Info
{
Configuration Config;
std::vector<std::string> BucketNames;
uint64_t EntryCount = 0;
uint64_t TotalSize = 0;
};
struct BucketStats
{
uint64_t TotalSize;
uint64_t HitCount;
uint64_t MissCount;
uint64_t WriteCount;
metrics::RequestStatsSnapshot PutOps;
metrics::RequestStatsSnapshot GetOps;
};
struct NamedBucketStats
{
std::string BucketName;
BucketStats Stats;
};
struct DiskStats
{
std::vector<NamedBucketStats> BucketStats;
};
explicit ZenCacheDiskLayer(const std::filesystem::path& RootDir, bool EnableReferenceCaching);
~ZenCacheDiskLayer();
bool Get(std::string_view Bucket, const IoHash& HashKey, ZenCacheValue& OutValue);
void Put(std::string_view Bucket, const IoHash& HashKey, const ZenCacheValue& Value, std::span<IoHash> References);
bool Drop();
bool DropBucket(std::string_view Bucket);
void Flush();
void ScrubStorage(ScrubContext& Ctx);
void GatherReferences(GcContext& GcCtx);
void CollectGarbage(GcContext& GcCtx);
void UpdateAccessTimes(const zen::access_tracking::AccessTimes& AccessTimes);
void DiscoverBuckets();
uint64_t TotalSize() const;
DiskStats Stats() const;
Info GetInfo() const;
std::optional<BucketInfo> GetBucketInfo(std::string_view Bucket) const;
void EnumerateBucketContents(std::string_view Bucket,
std::function<void(const IoHash& Key, const CacheValueDetails::ValueDetails& Details)>& Fn) const;
CacheValueDetails::NamespaceDetails GetValueDetails(const std::string_view BucketFilter, const std::string_view ValueFilter) const;
private:
/** A cache bucket manages a single directory containing
metadata and data for that bucket
*/
struct CacheBucket
{
CacheBucket(std::string BucketName, bool EnableReferenceCaching);
~CacheBucket();
bool OpenOrCreate(std::filesystem::path BucketDir, bool AllowCreate = true);
bool Get(const IoHash& HashKey, ZenCacheValue& OutValue);
void Put(const IoHash& HashKey, const ZenCacheValue& Value, std::span<IoHash> References);
bool Drop();
void Flush();
void ScrubStorage(ScrubContext& Ctx);
void GatherReferences(GcContext& GcCtx);
void CollectGarbage(GcContext& GcCtx);
void UpdateAccessTimes(const std::vector<zen::access_tracking::KeyAccessTime>& AccessTimes);
inline uint64_t TotalSize() const { return m_TotalStandaloneSize.load(std::memory_order::relaxed) + m_BlockStore.TotalSize(); }
uint64_t EntryCount() const;
BucketStats Stats();
CacheValueDetails::BucketDetails GetValueDetails(const std::string_view ValueFilter) const;
void EnumerateBucketContents(std::function<void(const IoHash& Key, const CacheValueDetails::ValueDetails& Details)>& Fn) const;
private:
const uint64_t MaxBlockSize = 1ull << 30;
uint64_t m_PayloadAlignment = 1ull << 4;
std::string m_BucketName;
std::filesystem::path m_BucketDir;
std::filesystem::path m_BlocksBasePath;
BlockStore m_BlockStore;
Oid m_BucketId;
uint64_t m_LargeObjectThreshold = 128 * 1024;
std::atomic_bool m_IsFlushing{};
const bool m_EnableReferenceCaching = false;
// These files are used to manage storage of small objects for this bucket
TCasLogFile<DiskIndexEntry> m_SlogFile;
uint64_t m_LogFlushPosition = 0;
#pragma pack(push)
#pragma pack(1)
static const size_t UnknownReferencesIndex = (size_t)-1;
static const size_t NoReferencesIndex = (size_t)-2;
struct BucketPayload
{
DiskLocation Location; // 12
uint64_t RawSize; // 8
IoHash RawHash; // 20
};
#pragma pack(pop)
static_assert(sizeof(BucketPayload) == 40u);
static_assert(sizeof(AccessTime) == 4u);
using IndexMap = tsl::robin_map<IoHash, size_t, IoHash::Hasher>;
std::atomic<uint64_t> m_HitCount;
std::atomic<uint64_t> m_MissCount;
std::atomic<uint64_t> m_WriteCount;
metrics::RequestStats m_PutOps;
metrics::RequestStats m_GetOps;
mutable RwLock m_IndexLock;
IndexMap m_Index;
std::vector<AccessTime> m_AccessTimes;
std::vector<BucketPayload> m_Payloads;
std::vector<size_t> m_FirstReferenceIndex;
std::vector<IoHash> m_ReferenceHashes;
std::vector<size_t> m_NextReferenceHashesIndexes;
size_t m_ReferenceCount = 0;
std::atomic_uint64_t m_TotalStandaloneSize{};
void BuildPath(PathBuilderBase& Path, const IoHash& HashKey) const;
void PutStandaloneCacheValue(const IoHash& HashKey, const ZenCacheValue& Value, std::span<IoHash> References);
IoBuffer GetStandaloneCacheValue(ZenContentType ContentType, const IoHash& HashKey) const;
void PutInlineCacheValue(const IoHash& HashKey, const ZenCacheValue& Value, std::span<IoHash> References);
IoBuffer GetInlineCacheValue(const DiskLocation& Loc) const;
void MakeIndexSnapshot();
uint64_t ReadIndexFile(const std::filesystem::path& IndexPath, uint32_t& OutVersion);
uint64_t ReadLog(const std::filesystem::path& LogPath, uint64_t LogPosition);
void OpenLog(const bool IsNew);
CbObject MakeManifest(IndexMap&& Index, std::vector<AccessTime>&& AccessTimes, const std::vector<BucketPayload>& Payloads);
void SaveManifest(CbObject&& Manifest);
CacheValueDetails::ValueDetails GetValueDetails(const IoHash& Key, size_t Index) const;
void CompactReferences(RwLock::ExclusiveLockScope&);
void SetReferences(RwLock::ExclusiveLockScope&, std::size_t& FirstReferenceIndex, std::span<IoHash> References);
void RemoveReferences(RwLock::ExclusiveLockScope&, std::size_t& FirstReferenceIndex);
inline bool GetReferences(RwLock::SharedLockScope&, std::size_t FirstReferenceIndex, std::vector<IoHash>& OutReferences) const
{
return LockedGetReferences(FirstReferenceIndex, OutReferences);
}
inline bool GetReferences(RwLock::ExclusiveLockScope&, std::size_t FirstReferenceIndex, std::vector<IoHash>& OutReferences) const
{
return LockedGetReferences(FirstReferenceIndex, OutReferences);
}
size_t AllocateReferenceEntry(RwLock::ExclusiveLockScope&, const IoHash& Key);
bool LockedGetReferences(std::size_t FirstReferenceIndex, std::vector<IoHash>& OutReferences) const;
// These locks are here to avoid contention on file creation, therefore it's sufficient
// that we take the same lock for the same hash
//
// These locks are small and should really be spaced out so they don't share cache lines,
// but we don't currently access them at particularly high frequency so it should not be
// an issue in practice
mutable RwLock m_ShardedLocks[256];
inline RwLock& LockForHash(const IoHash& Hash) const { return m_ShardedLocks[Hash.Hash[19]]; }
};
std::filesystem::path m_RootDir;
mutable RwLock m_Lock;
std::unordered_map<std::string, std::unique_ptr<CacheBucket>> m_Buckets; // TODO: make this case insensitive
std::vector<std::unique_ptr<CacheBucket>> m_DroppedBuckets;
const bool m_EnableReferenceCaching;
ZenCacheDiskLayer(const ZenCacheDiskLayer&) = delete;
ZenCacheDiskLayer& operator=(const ZenCacheDiskLayer&) = delete;
};
} // namespace zen
|