AP-2214 Fix crash on monitorAdsSlotChangesTask

(cherry picked from commit e10908a)
meganz · Oct 29, 2024 · 499156c · 499156c
1 parent 43e122d
commit 499156c
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/Modules/Presentation/Features/Accounts/Sources/Accounts/Ads/ViewModel/AdsSlotViewModel.swift b/Modules/Presentation/Features/Accounts/Sources/Accounts/Ads/ViewModel/AdsSlotViewModel.swift
@@ -29,11 +29,6 @@ final public class AdsSlotViewModel: ObservableObject {
         self.appEnvironmentUseCase = appEnvironmentUseCase
     }
 
-    deinit {
-        monitorAdsSlotChangesTask?.cancel()
-        monitorAdsSlotChangesTask = nil
-    }
-
     // MARK: Setup
     func setupSubscriptions() {
         NotificationCenter.default
@@ -64,11 +59,9 @@ final public class AdsSlotViewModel: ObservableObject {
 
     // MARK: Ads Slot changes
     func monitorAdsSlotChanges() {
-        monitorAdsSlotChangesTask = Task { [weak self] in
-            guard let self else { return }
+        monitorAdsSlotChangesTask = Task { [weak self, adsSlotChangeStream] in
             for await newAdsSlotConfig in adsSlotChangeStream.adsSlotStream {
-                if Task.isCancelled { return }
-                await updateAdsSlot(newAdsSlotConfig)
+                await self?.updateAdsSlot(newAdsSlotConfig)
             }
         }
     }