AP-2214: Add back the deinit method so the task can be cancelled

meganz · Oct 29, 2024 · 5b85cd0 · 5b85cd0
1 parent 499156c
commit 5b85cd0
Showing 1 changed file with 6 additions and 0 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,6 +29,11 @@ final public class AdsSlotViewModel: ObservableObject {
         self.appEnvironmentUseCase = appEnvironmentUseCase
     }
 
+    deinit {
+        monitorAdsSlotChangesTask?.cancel()
+        monitorAdsSlotChangesTask = nil
+    }
+
     // MARK: Setup
     func setupSubscriptions() {
         NotificationCenter.default
@@ -59,6 +64,7 @@ final public class AdsSlotViewModel: ObservableObject {
 
     // MARK: Ads Slot changes
     func monitorAdsSlotChanges() {
+        monitorAdsSlotChangesTask?.cancel()
         monitorAdsSlotChangesTask = Task { [weak self, adsSlotChangeStream] in
             for await newAdsSlotConfig in adsSlotChangeStream.adsSlotStream {
                 await self?.updateAdsSlot(newAdsSlotConfig)