Fixed bug in z$ GC.

1 files changed, 36 insertions, 34 deletions

diff --git a/zenserver/cache/structuredcachestore.cpp b/zenserver/cache/structuredcachestore.cpp
index 030588659..44c7b728b 100644
--- a/zenserver/cache/structuredcachestore.cpp
+++ b/zenserver/cache/structuredcachestore.cpp
@@ -444,7 +444,7 @@ static_assert(sizeof(DiskIndexEntry) == 36);
 
 struct ZenCacheDiskLayer::CacheBucket
 {
-	CacheBucket();
+	CacheBucket(std::string BucketName);
 	~CacheBucket();
 
 	void		OpenOrCreate(std::filesystem::path BucketDir, bool AllowCreate = true);
@@ -463,6 +463,7 @@ struct ZenCacheDiskLayer::CacheBucket
 	inline uint64_t TotalSize() const { return m_TotalSize.load(std::memory_order::relaxed); }
 
 private:
+	std::string			  m_BucketName;
 	std::filesystem::path m_BucketDir;
 	Oid					  m_BucketId;
 	bool				  m_IsOk				 = false;
@@ -504,7 +505,7 @@ private:
 	inline RwLock& LockForHash(const IoHash& Hash) { return m_ShardedLocks[Hash.Hash[19]]; }
 };
 
-ZenCacheDiskLayer::CacheBucket::CacheBucket()
+ZenCacheDiskLayer::CacheBucket::CacheBucket(std::string BucketName) : m_BucketName(std::move(BucketName))
 {
 }
 
@@ -916,7 +917,7 @@ ZenCacheDiskLayer::CacheBucket::GatherReferences(GcContext& GcCtx)
 	std::transform(Entries.begin(), ValidIt, std::back_inserter(ExpiredKeys), [](const auto& Kv) { return Kv.first; });
 
 	GcCtx.ContributeCids(Cids);
-	GcCtx.ContributeCacheKeys(std::move(ValidKeys), std::move(ExpiredKeys));
+	GcCtx.ContributeCacheKeys(m_BucketName, std::move(ValidKeys), std::move(ExpiredKeys));
 }
 
 void
@@ -963,8 +964,8 @@ ZenCacheDiskLayer::CacheBucket::CollectGarbage(GcContext& GcCtx)
 	std::vector<IndexMap::value_type> ValidEntries;
 	std::vector<IndexMap::value_type> ExpiredEntries;
 
-	AddEntries(GcCtx.ValidCacheKeys(), ValidEntries);
-	AddEntries(GcCtx.ExpiredCacheKeys(), ExpiredEntries);
+	AddEntries(GcCtx.ValidCacheKeys(m_BucketName), ValidEntries);
+	AddEntries(GcCtx.ExpiredCacheKeys(m_BucketName), ExpiredEntries);
 
 	// Remove all standalone file(s)
 	// NOTE: This can probably be made asynchronously
@@ -1268,12 +1269,13 @@ ZenCacheDiskLayer::~ZenCacheDiskLayer() = default;
 bool
 ZenCacheDiskLayer::Get(std::string_view InBucket, const IoHash& HashKey, ZenCacheValue& OutValue)
 {
-	CacheBucket* Bucket = nullptr;
+	const auto	 BucketName = std::string(InBucket);
+	CacheBucket* Bucket		= nullptr;
 
 	{
 		RwLock::SharedLockScope _(m_Lock);
 
-		auto it = m_Buckets.find(std::string(InBucket));
+		auto it = m_Buckets.find(BucketName);
 
 		if (it != m_Buckets.end())
 		{
@@ -1287,17 +1289,17 @@ ZenCacheDiskLayer::Get(std::string_view InBucket, const IoHash& HashKey, ZenCach
 
 		RwLock::ExclusiveLockScope _(m_Lock);
 
-		if (auto it = m_Buckets.find(std::string(InBucket)); it != m_Buckets.end())
+		if (auto it = m_Buckets.find(BucketName); it != m_Buckets.end())
 		{
 			Bucket = &it->second;
 		}
 		else
 		{
-			auto It = m_Buckets.try_emplace(std::string(InBucket));
+			auto It = m_Buckets.try_emplace(BucketName, BucketName);
 			Bucket	= &It.first->second;
 
 			std::filesystem::path BucketPath = m_RootDir;
-			BucketPath /= std::string(InBucket);
+			BucketPath /= BucketName;
 
 			Bucket->OpenOrCreate(BucketPath);
 		}
@@ -1311,12 +1313,13 @@ ZenCacheDiskLayer::Get(std::string_view InBucket, const IoHash& HashKey, ZenCach
 void
 ZenCacheDiskLayer::Put(std::string_view InBucket, const IoHash& HashKey, const ZenCacheValue& Value)
 {
-	CacheBucket* Bucket = nullptr;
+	const auto	 BucketName = std::string(InBucket);
+	CacheBucket* Bucket		= nullptr;
 
 	{
 		RwLock::SharedLockScope _(m_Lock);
 
-		auto it = m_Buckets.find(std::string(InBucket));
+		auto it = m_Buckets.find(BucketName);
 
 		if (it != m_Buckets.end())
 		{
@@ -1330,17 +1333,17 @@ ZenCacheDiskLayer::Put(std::string_view InBucket, const IoHash& HashKey, const Z
 
 		RwLock::ExclusiveLockScope _(m_Lock);
 
-		if (auto it = m_Buckets.find(std::string(InBucket)); it != m_Buckets.end())
+		if (auto it = m_Buckets.find(BucketName); it != m_Buckets.end())
 		{
 			Bucket = &it->second;
 		}
 		else
 		{
-			auto It = m_Buckets.try_emplace(std::string(InBucket));
+			auto It = m_Buckets.try_emplace(BucketName, BucketName);
 			Bucket	= &It.first->second;
 
 			std::filesystem::path bucketPath = m_RootDir;
-			bucketPath /= std::string(InBucket);
+			bucketPath /= BucketName;
 
 			Bucket->OpenOrCreate(bucketPath);
 		}
@@ -1392,7 +1395,7 @@ ZenCacheDiskLayer::DiscoverBuckets()
 		}
 		else
 		{
-			auto InsertResult = m_Buckets.try_emplace(BucketName8);
+			auto InsertResult = m_Buckets.try_emplace(BucketName8, BucketName8);
 
 			std::filesystem::path BucketPath = m_RootDir;
 			BucketPath /= BucketName8;
@@ -1504,7 +1507,6 @@ using namespace std::literals;
 using namespace fmt::literals;
 
 namespace testutils {
-
 	IoHash CreateKey(size_t KeyValue) { return IoHash::HashBuffer(&KeyValue, sizeof(size_t)); }
 
 	IoBuffer CreateBinaryCacheValue(uint64_t Size)
@@ -1715,23 +1717,23 @@ TEST_CASE("z$.gc")
 		}
 
 		// Expect timestamps to be serialized
-		//{
-		//	CasGc				Gc;
-		//	ZenCacheStore		Zcs(Gc, TempDir.Path() / "cache");
-		//	std::vector<IoHash> Keep;
-
-		//	// Collect garbage with 1 hour max cache duration
-		//	{
-		//		CollectAndFilter(Gc, GcClock::Now(), std::chrono::hours(1), Cids, Keep);
-		//		CHECK_EQ(3, Keep.size());
-		//	}
-
-		//	// Move forward in time
-		//	{
-		//		CollectAndFilter(Gc, GcClock::Now() + std::chrono::hours(2), std::chrono::hours(1), Cids, Keep);
-		//		CHECK_EQ(0, Keep.size());
-		//	}
-		//}
+		{
+			CasGc				Gc;
+			ZenCacheStore		Zcs(Gc, TempDir.Path() / "cache");
+			std::vector<IoHash> Keep;
+
+			// Collect garbage with 1 hour max cache duration
+			{
+				CollectAndFilter(Gc, GcClock::Now(), std::chrono::hours(1), Cids, Keep);
+				CHECK_EQ(3, Keep.size());
+			}
+
+			// Move forward in time
+			{
+				CollectAndFilter(Gc, GcClock::Now() + std::chrono::hours(2), std::chrono::hours(1), Cids, Keep);
+				CHECK_EQ(0, Keep.size());
+			}
+		}
 	}
 
 	SUBCASE("gc removes standalone values")