Make sure we close all block files when dropping a cache bucket

2 files changed, 12 insertions, 0 deletions

diff --git a/zenstore/blockstore.cpp b/zenstore/blockstore.cpp
index e502113fc..b82b93823 100644
--- a/zenstore/blockstore.cpp
+++ b/zenstore/blockstore.cpp
@@ -193,6 +193,17 @@ BlockStore::Initialize(const std::filesystem::path&			  BlocksBasePath,
 	}
 }
 
+void
+BlockStore::Close()
+{
+	RwLock::ExclusiveLockScope InsertLock(m_InsertLock);
+	m_WriteBlock		  = nullptr;
+	m_CurrentInsertOffset = 0;
+	m_WriteBlockIndex	  = 0;
+	m_ChunkBlocks.clear();
+	m_BlocksBasePath.clear();
+}
+
 BlockStoreLocation
 BlockStore::WriteChunk(const void* Data, uint64_t Size, uint64_t Alignment)
 {
diff --git a/zenstore/include/zenstore/blockstore.h b/zenstore/include/zenstore/blockstore.h
index 5019e257d..e153b530e 100644
--- a/zenstore/include/zenstore/blockstore.h
+++ b/zenstore/include/zenstore/blockstore.h
@@ -131,6 +131,7 @@ public:
 					uint64_t							   MaxBlockSize,
 					uint64_t							   MaxBlockCount,
 					const std::vector<BlockStoreLocation>& KnownLocations);
+	void Close();
 
 	BlockStoreLocation WriteChunk(const void* Data, uint64_t Size, uint64_t Alignment);