Add possibility to disable scheduling of background tasks

DP-3T · Mar 9, 2022 · 460c357 · 460c357
1 parent 998034c
commit 460c357
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/Sources/DP3TSDK/Background/DP3TBackgroundTaskManager.swift b/Sources/DP3TSDK/Background/DP3TBackgroundTaskManager.swift
@@ -31,6 +31,8 @@ class DP3TBackgroundTaskManager {
     private let serviceClient: ExposeeServiceClientProtocol
 
     private weak var tracer: Tracer!
+
+    var shouldReschedule: Bool = true
 
     init(handler: DP3TBackgroundHandler?,
          keyProvider: DiagnosisKeysProvider,
@@ -183,6 +185,8 @@ class DP3TBackgroundTaskManager {
 
     @available(iOS 13.0, *)
     private func scheduleBackgroundTasks() {
+        guard self.shouldReschedule else { return }
+
         logger.trace()
 
         // Schedule next app refresh task 12h in the future

diff --git a/Sources/DP3TSDK/DP3TSDK.swift b/Sources/DP3TSDK/DP3TSDK.swift
@@ -375,6 +375,10 @@ class DP3TSDK {
         exposureDayStorage.reset()
         URLCache.shared.removeAllCachedResponses()
     }
+
+    func setBackgroundTasksEnabled(_ enabled: Bool) {
+        backgroundTaskManager.shouldReschedule = enabled
+    }
 
     @objc func backgroundRefreshStatusDidChange() {
         let new = UIApplication.shared.backgroundRefreshStatus

diff --git a/Sources/DP3TSDK/DP3TTracing.swift b/Sources/DP3TSDK/DP3TTracing.swift
@@ -206,6 +206,12 @@ public enum DP3TTracing {
         instance = nil
     }
 
+    @available(iOS 12.5, *)
+    public static func setBackgroundTasksEnabled(_ enabled: Bool) {
+        instancePrecondition()
+        instance.setBackgroundTasksEnabled(enabled)
+    }
+
     public static var loggingEnabled: Bool {
         set {
             Logger.loggingEnabled = newValue