add ChunkService.GetAllChunkFilesInDirectory method

Etherna · Jul 29, 2024 · 5a9041a · 5a9041a
1 parent bdc4ce2
commit 5a9041a
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/src/BeeNet.Util/Hashing/Store/LocalDirectoryChunkStore.cs b/src/BeeNet.Util/Hashing/Store/LocalDirectoryChunkStore.cs
@@ -29,7 +29,7 @@ namespace Etherna.BeeNet.Hashing.Store
     public class LocalDirectoryChunkStore : IChunkStore
     {
         // Consts.
-        const string ChunkFileExtension = ".chunk";
+        public const string ChunkFileExtension = ".chunk";
 
         // Constructor.
         public LocalDirectoryChunkStore(string directoryPath, bool createDirectory = false)

diff --git a/src/BeeNet.Util/Services/ChunkService.cs b/src/BeeNet.Util/Services/ChunkService.cs
@@ -235,6 +235,9 @@ public async Task<UploadEvaluationResult> EvaluateSingleFileUploadAsync(
                 postageStampIssuer);
         }
 
+        public IEnumerable<string> GetAllChunkFilesInDirectory(string chunkStoreDirectory) =>
+            Directory.GetFiles(chunkStoreDirectory, '*' + LocalDirectoryChunkStore.ChunkFileExtension);
+
         public async Task<IReadOnlyDictionary<string, string>> GetFileMetadataFromChunksAsync(
             string chunkStoreDirectory,
             SwarmAddress address)

diff --git a/src/BeeNet.Util/Services/IChunkService.cs b/src/BeeNet.Util/Services/IChunkService.cs
@@ -96,6 +96,14 @@ Task<UploadEvaluationResult> EvaluateSingleFileUploadAsync(
             int? chunkCuncorrency = null,
             IChunkStore? chunkStore = null);
 
+        /// <summary>
+        /// Get list of all chunk files in directory
+        /// </summary>
+        /// <param name="chunkStoreDirectory">The chunk directory</param>
+        /// <returns>List of chunk files</returns>
+        IEnumerable<string> GetAllChunkFilesInDirectory(
+            string chunkStoreDirectory);
+
         /// <summary>
         /// Get resource metadata from a directory of chunks and the resource address
         /// </summary>