Keep a ref to store in AnyStore classes (#22)

omaralbeik · Jul 30, 2023 · 6382917 · 6382917
1 parent b835f7e
commit 6382917
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/Sources/Blueprints/AnyMultiObjectStore.swift b/Sources/Blueprints/AnyMultiObjectStore.swift
@@ -7,6 +7,7 @@ public final class AnyMultiObjectStore<
   public init<Store: MultiObjectStore>(
     _ store: Store
   ) where Store.Object == Object {
+    _store = store
     _save = { try store.save($0) }
     _saveOptional = { try store.save($0) }
     _saveObjects = { try store.save($0) }
@@ -20,6 +21,7 @@ public final class AnyMultiObjectStore<
     _removeAll = { try store.removeAll() }
   }
 
+  private let _store: any MultiObjectStore
   private let _save: (Object) throws -> Void
   private let _saveOptional: (Object?) throws -> Void
   private let _saveObjects: ([Object]) throws -> Void

diff --git a/Sources/Blueprints/AnySingleObjectStore.swift b/Sources/Blueprints/AnySingleObjectStore.swift
@@ -5,12 +5,14 @@ public final class AnySingleObjectStore<Object: Codable>: SingleObjectStore {
   public init<Store: SingleObjectStore>(
     _ store: Store
   ) where Store.Object == Object {
+    _store = store
     _save = { try store.save($0) }
     _saveOptional = { try store.save($0) }
     _object = { store.object() }
     _remove = { try store.remove() }
   }
 
+  private let _store: any SingleObjectStore
   private let _save: (Object) throws -> Void
   private let _saveOptional: (Object?) throws -> Void
   private let _object: () -> Object?