Merge pull request #280 from asam139/feature/all_keys_and_objects

Added allKeys and allObjects computed vars for storage with sense

Author: 3lvis

Source/Shared/Storage/DiskStorage.swift

@@ -58,6 +58,10 @@ final public class DiskStorage<Key: Hashable, Value> {
 }
 
 extension DiskStorage: StorageAware {
+  public var allKeys: [Key] { [] }
+
+  public var allObjects: [Value] { [] }
+
   public func entry(forKey key: Key) throws -> Entry<Value> {
     let filePath = makeFilePath(for: key)
     let data = try Data(contentsOf: URL(fileURLWithPath: filePath))

Source/Shared/Storage/HybridStorage.swift

@@ -26,6 +26,14 @@ public final class HybridStorage<Key: Hashable, Value> {
 }
 
 extension HybridStorage: StorageAware {
+  public var allKeys: [Key] {
+    memoryStorage.allKeys
+  }
+
+  public var allObjects: [Value] {
+    memoryStorage.allObjects
+  }
+
   public func entry(forKey key: Key) throws -> Entry<Value> {
     do {
       return try memoryStorage.entry(forKey: key)

Source/Shared/Storage/MemoryStorage.swift

@@ -31,6 +31,14 @@ public class MemoryStorage<Key: Hashable, Value>: StorageAware {
 }
 
 extension MemoryStorage {
+  public var allKeys: [Key] {
+    Array(keys)
+  }
+
+  public var allObjects: [Value] {
+    allKeys.compactMap { try? object(forKey: $0) }
+  }
+
   public func setObject(_ object: Value, forKey key: Key, expiry: Expiry? = nil) {
     let capsule = MemoryCapsule(value: object, expiry: .date(expiry?.date ?? config.expiry.date))
     cache.setObject(capsule, forKey: WrappedKey(key))

Source/Shared/Storage/Storage.swift

@@ -43,6 +43,14 @@ public final class Storage<Key: Hashable, Value> {
 }
 
 extension Storage: StorageAware {
+  public var allKeys: [Key] {
+    self.syncStorage.allKeys
+  }
+
+  public var allObjects: [Value] {
+    self.syncStorage.allObjects
+  }
+
   public func entry(forKey key: Key) throws -> Entry<Value> {
     return try self.syncStorage.entry(forKey: key)
   }

Source/Shared/Storage/StorageAware.swift

@@ -4,6 +4,17 @@ import Foundation
 public protocol StorageAware {
   associatedtype Key: Hashable
   associatedtype Value
+
+  /**
+   Get all keys in the storage
+  */
+  var allKeys: [Key] { get }
+
+  /**
+   Get all objects from the storage
+  */
+  var allObjects: [Value] { get }
+  
   /**
    Tries to retrieve the object from the storage.
    - Parameter key: Unique key to identify the object in the cache

Source/Shared/Storage/SyncStorage.swift

@@ -14,12 +14,27 @@ public class SyncStorage<Key: Hashable, Value> {
 }
 
 extension SyncStorage: StorageAware {
+  public var allKeys: [Key] {
+    var keys: [Key]!
+    serialQueue.sync {
+      keys = self.innerStorage.allKeys
+    }
+    return keys
+  }
+
+  public var allObjects: [Value] {
+    var objects: [Value]!
+    serialQueue.sync {
+      objects = self.innerStorage.allObjects
+    }
+    return objects
+  }
+
   public func entry(forKey key: Key) throws -> Entry<Value> {
     var entry: Entry<Value>!
     try serialQueue.sync {
       entry = try innerStorage.entry(forKey: key)
     }
-
     return entry
   }