fix

Ryu0118 · Oct 16, 2023 · be62d4f · be62d4f
1 parent bd1010d
commit be62d4f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/Sources/SimplexArchitecture/StateContainer.swift b/Sources/SimplexArchitecture/StateContainer.swift
@@ -13,6 +13,8 @@ public final class StateContainer<Target: ActionSendable> {
     @TestOnly var viewState: Target.ViewState?
 
     var _reducerState: Target.Reducer.ReducerState?
+
+    @usableFromInline
     var entity: Target
 
     init(

diff --git a/Sources/SimplexArchitecture/Store/Store.swift b/Sources/SimplexArchitecture/Store/Store.swift
@@ -69,6 +69,7 @@ public final class Store<Reducer: ReducerProtocol> {
         viewState: Reducer.Target.ViewState? = nil
     ) -> StateContainer<Reducer.Target> {
         if let container {
+            container.entity = target
             return container
         } else {
             let container = getContainer(for: target, viewState: viewState)