Fixed build list performance issue

sergiikhliustin · Jan 24, 2023 · c054c88 · c054c88
1 parent 961a9ab
commit c054c88
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/Modules/BuildioLogic/Sources/BuildioLogic/Factory/ViewModelFactory.swift b/Modules/BuildioLogic/Sources/BuildioLogic/Factory/ViewModelFactory.swift
@@ -62,12 +62,11 @@ public final class ViewModelFactory {
     }
 
     public func build(_ build: BuildResponseItemModel) -> BuildViewModel {
-        // TODO: Double check
-        let model = cached(key: "BuildViewModel_\(build.slug)", model: BuildViewModel(tokenManager, build: build))
-        if build.status != .running {
-            model.value = build
+        if build.status == .running {
+            return cached(key: "BuildViewModel_\(build.slug)", model: BuildViewModel(tokenManager, build: build))
+        } else {
+            return BuildViewModel(tokenManager, build: build)
         }
-        return model
     }
 
     public func logs(_ build: BuildResponseItemModel) -> LogsViewModel {