Ensure state is updated on MainActor (#20)

* Run task on main actor * No need for this
reddavis · May 3, 2022 · 83e2a2b · 83e2a2b
1 parent 9be06fa
commit 83e2a2b
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/RedUx/Source/ViewModel.swift b/RedUx/Source/ViewModel.swift
@@ -56,13 +56,15 @@ public final class ViewModel<State: Equatable, Event>: ObservableObject {
         self.state = store.state
         self._send = { store.send($0) }
         self.stateTask = Task { [weak self] in
+            guard let self = self else { return }
+
             do {
-                self?.state = store.state
+                await MainActor.run {
+                    self.state = store.state
+                }
 
                 for try await state in store.stateSequence.removeDuplicates() {
-                    guard
-                        let self = self,
-                        !Task.isCancelled else { break }
+                    guard !Task.isCancelled else { break }
 
                     await MainActor.run {
                         self.state = state