explicitly cancel BGTasks

DP-3T · Mar 17, 2022 · bd10796 · bd10796
1 parent 97997e5
commit bd10796
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/Sources/DP3TSDK/Background/DP3TBackgroundTaskManager.swift b/Sources/DP3TSDK/Background/DP3TBackgroundTaskManager.swift
@@ -32,7 +32,15 @@ class DP3TBackgroundTaskManager {
 
     private weak var tracer: Tracer!
 
-    var shouldReschedule: Bool = true
+    var shouldReschedule: Bool = true {
+        didSet {
+            if shouldReschedule == false {
+                if #available(iOS 13.0, *) {
+                    cancelScheduledTask()
+                }
+            }
+        }
+    }
 
     init(handler: DP3TBackgroundHandler?,
          keyProvider: DiagnosisKeysProvider,
@@ -87,6 +95,11 @@ class DP3TBackgroundTaskManager {
 
     private func handleiOS12BackgroundLaunch() {
         logger.trace()
+
+        if !shouldReschedule {
+            return
+        }
+
         let queue = OperationQueue()
 
         if let handler = handler {
@@ -110,6 +123,10 @@ class DP3TBackgroundTaskManager {
     private func handleExposureNotificationBackgroundTask(_ task: BGTask) {
         logger.trace()
         scheduleBackgroundTasks()
+
+        if !shouldReschedule {
+            return
+        }
 
         let queue = OperationQueue()
 
@@ -213,4 +230,9 @@ class DP3TBackgroundTaskManager {
             logger.error("Exposure notification task schedule failed error: %{public}@", error.localizedDescription)
         }
     }
+
+    @available(iOS 13.0, *)
+    func cancelScheduledTask() {
+        BGTaskScheduler.shared.cancelAllTaskRequests()
+    }
 }