|
|
|
@ -23,7 +23,11 @@ public enum SessionBackgroundTaskState {
|
|
|
|
// MARK: - SessionBackgroundTaskManager
|
|
|
|
// MARK: - SessionBackgroundTaskManager
|
|
|
|
|
|
|
|
|
|
|
|
public class SessionBackgroundTaskManager {
|
|
|
|
public class SessionBackgroundTaskManager {
|
|
|
|
|
|
|
|
/// Maximum duration to extend background tasks
|
|
|
|
|
|
|
|
private static let maxBackgroundTime: TimeInterval = 180
|
|
|
|
|
|
|
|
|
|
|
|
private let dependencies: Dependencies
|
|
|
|
private let dependencies: Dependencies
|
|
|
|
|
|
|
|
private let queue = DispatchQueue(label: "com.session.backgroundTaskManager")
|
|
|
|
|
|
|
|
|
|
|
|
/// This property should only be accessed while synchronized on this instance.
|
|
|
|
/// This property should only be accessed while synchronized on this instance.
|
|
|
|
private var backgroundTaskId: UIBackgroundTaskIdentifier = .invalid
|
|
|
|
private var backgroundTaskId: UIBackgroundTaskIdentifier = .invalid
|
|
|
|
@ -43,11 +47,11 @@ public class SessionBackgroundTaskManager {
|
|
|
|
/// begins, we use a single uninterrupted background that spans their lifetimes.
|
|
|
|
/// begins, we use a single uninterrupted background that spans their lifetimes.
|
|
|
|
///
|
|
|
|
///
|
|
|
|
/// This property should only be accessed while synchronized on this instance.
|
|
|
|
/// This property should only be accessed while synchronized on this instance.
|
|
|
|
private var continuityTimer: Timer?
|
|
|
|
private var continuityTimer: DispatchSourceTimer?
|
|
|
|
|
|
|
|
|
|
|
|
/// In order to ensure we have sufficient time to clean up before background tasks expire (without having to kick off additional tasks)
|
|
|
|
/// In order to ensure we have sufficient time to clean up before background tasks expire (without having to kick off additional tasks)
|
|
|
|
/// we track the remaining background execution time and end tasks 5 seconds early (same as the AppDelegate background fetch)
|
|
|
|
/// we track the remaining background execution time and end tasks 5 seconds early (same as the AppDelegate background fetch)
|
|
|
|
private var expirationTimeObserver: Timer?
|
|
|
|
private var expirationTimeObserver: DispatchSourceTimer?
|
|
|
|
private var hasGottenValidBackgroundTimeRemaining: Bool = false
|
|
|
|
private var hasGottenValidBackgroundTimeRemaining: Bool = false
|
|
|
|
|
|
|
|
|
|
|
|
fileprivate init(using dependencies: Dependencies) {
|
|
|
|
fileprivate init(using dependencies: Dependencies) {
|
|
|
|
@ -61,13 +65,6 @@ public class SessionBackgroundTaskManager {
|
|
|
|
|
|
|
|
|
|
|
|
// MARK: - Functions
|
|
|
|
// MARK: - Functions
|
|
|
|
|
|
|
|
|
|
|
|
@discardableResult private static func synced<T>(_ lock: Any, closure: () -> T) -> T {
|
|
|
|
|
|
|
|
objc_sync_enter(lock)
|
|
|
|
|
|
|
|
let result: T = closure()
|
|
|
|
|
|
|
|
objc_sync_exit(lock)
|
|
|
|
|
|
|
|
return result
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
public func startObservingNotifications() {
|
|
|
|
public func startObservingNotifications() {
|
|
|
|
guard dependencies[singleton: .appContext].isMainApp else { return }
|
|
|
|
guard dependencies[singleton: .appContext].isMainApp else { return }
|
|
|
|
|
|
|
|
|
|
|
|
@ -86,14 +83,14 @@ public class SessionBackgroundTaskManager {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
@objc private func applicationDidBecomeActive() {
|
|
|
|
@objc private func applicationDidBecomeActive() {
|
|
|
|
SessionBackgroundTaskManager.synced(self) { [weak self] in
|
|
|
|
queue.sync { [weak self] in
|
|
|
|
self?.isAppActive = true
|
|
|
|
self?.isAppActive = true
|
|
|
|
self?.ensureBackgroundTaskState()
|
|
|
|
self?.ensureBackgroundTaskState()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
@objc private func applicationWillResignActive() {
|
|
|
|
@objc private func applicationWillResignActive() {
|
|
|
|
SessionBackgroundTaskManager.synced(self) { [weak self] in
|
|
|
|
queue.sync { [weak self] in
|
|
|
|
self?.isAppActive = false
|
|
|
|
self?.isAppActive = false
|
|
|
|
self?.ensureBackgroundTaskState()
|
|
|
|
self?.ensureBackgroundTaskState()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
@ -109,7 +106,7 @@ public class SessionBackgroundTaskManager {
|
|
|
|
// background task, but the background task couldn't be begun.
|
|
|
|
// background task, but the background task couldn't be begun.
|
|
|
|
// In that case expirationBlock will not be called.
|
|
|
|
// In that case expirationBlock will not be called.
|
|
|
|
fileprivate func addTask(expiration: @escaping () -> ()) -> UInt64? {
|
|
|
|
fileprivate func addTask(expiration: @escaping () -> ()) -> UInt64? {
|
|
|
|
return SessionBackgroundTaskManager.synced(self) { [weak self, dependencies] in
|
|
|
|
return queue.sync { [weak self] () -> UInt64? in
|
|
|
|
let taskId: UInt64 = ((self?.idCounter ?? 0) + 1)
|
|
|
|
let taskId: UInt64 = ((self?.idCounter ?? 0) + 1)
|
|
|
|
self?.idCounter = taskId
|
|
|
|
self?.idCounter = taskId
|
|
|
|
self?.expirationMap[taskId] = expiration
|
|
|
|
self?.expirationMap[taskId] = expiration
|
|
|
|
@ -118,18 +115,15 @@ public class SessionBackgroundTaskManager {
|
|
|
|
self?.expirationMap.removeValue(forKey: taskId)
|
|
|
|
self?.expirationMap.removeValue(forKey: taskId)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
self?.continuityTimer?.invalidate()
|
|
|
|
if self?.continuityTimer != nil {
|
|
|
|
|
|
|
|
self?.continuityTimer?.cancel()
|
|
|
|
self?.continuityTimer = nil
|
|
|
|
self?.continuityTimer = nil
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// Start observing the background time remaining
|
|
|
|
// Start observing the background time remaining
|
|
|
|
if self?.expirationTimeObserver?.isValid != true {
|
|
|
|
if self?.expirationTimeObserver?.isCancelled == true {
|
|
|
|
self?.hasGottenValidBackgroundTimeRemaining = false
|
|
|
|
self?.hasGottenValidBackgroundTimeRemaining = false
|
|
|
|
self?.expirationTimeObserver = Timer.scheduledTimerOnMainThread(
|
|
|
|
self?.checkExpirationTime(in: .seconds(1)) // Don't know the remaining time so check soon
|
|
|
|
withTimeInterval: 1,
|
|
|
|
|
|
|
|
repeats: true,
|
|
|
|
|
|
|
|
using: dependencies,
|
|
|
|
|
|
|
|
block: { _ in self?.expirationTimerDidFire() }
|
|
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
return taskId
|
|
|
|
return taskId
|
|
|
|
@ -139,7 +133,7 @@ public class SessionBackgroundTaskManager {
|
|
|
|
fileprivate func removeTask(taskId: UInt64?) {
|
|
|
|
fileprivate func removeTask(taskId: UInt64?) {
|
|
|
|
guard let taskId: UInt64 = taskId else { return }
|
|
|
|
guard let taskId: UInt64 = taskId else { return }
|
|
|
|
|
|
|
|
|
|
|
|
SessionBackgroundTaskManager.synced(self) { [weak self, dependencies] in
|
|
|
|
queue.sync { [weak self, queue] in
|
|
|
|
self?.expirationMap.removeValue(forKey: taskId)
|
|
|
|
self?.expirationMap.removeValue(forKey: taskId)
|
|
|
|
|
|
|
|
|
|
|
|
// This timer will ensure that we keep the background task active (if necessary)
|
|
|
|
// This timer will ensure that we keep the background task active (if necessary)
|
|
|
|
@ -148,47 +142,54 @@ public class SessionBackgroundTaskManager {
|
|
|
|
// should be able to ensure background tasks by "narrowly" wrapping
|
|
|
|
// should be able to ensure background tasks by "narrowly" wrapping
|
|
|
|
// their core logic with a SessionBackgroundTask and not worrying about "hand off"
|
|
|
|
// their core logic with a SessionBackgroundTask and not worrying about "hand off"
|
|
|
|
// between SessionBackgroundTasks.
|
|
|
|
// between SessionBackgroundTasks.
|
|
|
|
self?.continuityTimer?.invalidate()
|
|
|
|
if self?.continuityTimer != nil {
|
|
|
|
self?.continuityTimer = Timer.scheduledTimerOnMainThread(
|
|
|
|
self?.continuityTimer?.cancel()
|
|
|
|
withTimeInterval: 0.25,
|
|
|
|
self?.continuityTimer = nil
|
|
|
|
using: dependencies,
|
|
|
|
}
|
|
|
|
block: { _ in self?.continuityTimerDidFire() }
|
|
|
|
|
|
|
|
)
|
|
|
|
self?.continuityTimer = DispatchSource.makeTimerSource(queue: queue)
|
|
|
|
|
|
|
|
self?.continuityTimer?.schedule(deadline: .now() + .milliseconds(250))
|
|
|
|
|
|
|
|
self?.continuityTimer?.setEventHandler { self?.continuityTimerDidFire() }
|
|
|
|
|
|
|
|
self?.continuityTimer?.resume()
|
|
|
|
self?.ensureBackgroundTaskState()
|
|
|
|
self?.ensureBackgroundTaskState()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/// Begins or end a background task if necessary.
|
|
|
|
/// Begins or end a background task if necessary
|
|
|
|
|
|
|
|
///
|
|
|
|
|
|
|
|
/// **Note:** Should only be called internally within `queue.sync` for thread safety
|
|
|
|
@discardableResult private func ensureBackgroundTaskState() -> Bool {
|
|
|
|
@discardableResult private func ensureBackgroundTaskState() -> Bool {
|
|
|
|
// We can't create background tasks in the SAE, but pretend that we succeeded.
|
|
|
|
// We can't create background tasks in the SAE, but pretend that we succeeded.
|
|
|
|
guard dependencies[singleton: .appContext].isMainApp else { return true }
|
|
|
|
guard dependencies[singleton: .appContext].isMainApp else { return true }
|
|
|
|
|
|
|
|
|
|
|
|
return SessionBackgroundTaskManager.synced(self) { [weak self, dependencies] in
|
|
|
|
|
|
|
|
// We only want to have a background task if we are:
|
|
|
|
// We only want to have a background task if we are:
|
|
|
|
// a) "not active" AND
|
|
|
|
// a) "not active" AND
|
|
|
|
// b1) there is one or more active instance of SessionBackgroundTask OR...
|
|
|
|
// b1) there is one or more active instance of SessionBackgroundTask OR...
|
|
|
|
// b2) ...there _was_ an active instance recently.
|
|
|
|
// b2) ...there _was_ an active instance recently.
|
|
|
|
let shouldHaveBackgroundTask: Bool = (
|
|
|
|
let shouldHaveBackgroundTask: Bool = (
|
|
|
|
self?.isAppActive == false && (
|
|
|
|
self.isAppActive == false && (
|
|
|
|
(self?.expirationMap.count ?? 0) > 0 ||
|
|
|
|
self.expirationMap.count > 0 ||
|
|
|
|
self?.continuityTimer != nil
|
|
|
|
self.continuityTimer != nil
|
|
|
|
)
|
|
|
|
)
|
|
|
|
)
|
|
|
|
)
|
|
|
|
let hasBackgroundTask: Bool = (self?.backgroundTaskId != .invalid)
|
|
|
|
let hasBackgroundTask: Bool = (self.backgroundTaskId != .invalid)
|
|
|
|
|
|
|
|
|
|
|
|
guard shouldHaveBackgroundTask != hasBackgroundTask else {
|
|
|
|
guard shouldHaveBackgroundTask != hasBackgroundTask else {
|
|
|
|
// Current state is correct
|
|
|
|
// Current state is correct
|
|
|
|
return true
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
guard !shouldHaveBackgroundTask else {
|
|
|
|
guard !shouldHaveBackgroundTask else {
|
|
|
|
return (self?.startBackgroundTask() == true)
|
|
|
|
return (self.startOverarchingBackgroundTask() == true)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// Need to end background task.
|
|
|
|
// Need to end background task.
|
|
|
|
let maybeBackgroundTaskId: UIBackgroundTaskIdentifier? = self?.backgroundTaskId
|
|
|
|
let maybeBackgroundTaskId: UIBackgroundTaskIdentifier? = self.backgroundTaskId
|
|
|
|
self?.backgroundTaskId = .invalid
|
|
|
|
self.backgroundTaskId = .invalid
|
|
|
|
self?.expirationTimeObserver?.invalidate()
|
|
|
|
|
|
|
|
self?.expirationTimeObserver = nil
|
|
|
|
if self.expirationTimeObserver != nil {
|
|
|
|
|
|
|
|
self.expirationTimeObserver?.cancel()
|
|
|
|
|
|
|
|
self.expirationTimeObserver = nil
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if let backgroundTaskId: UIBackgroundTaskIdentifier = maybeBackgroundTaskId, backgroundTaskId != .invalid {
|
|
|
|
if let backgroundTaskId: UIBackgroundTaskIdentifier = maybeBackgroundTaskId, backgroundTaskId != .invalid {
|
|
|
|
dependencies[singleton: .appContext].endBackgroundTask(backgroundTaskId)
|
|
|
|
dependencies[singleton: .appContext].endBackgroundTask(backgroundTaskId)
|
|
|
|
@ -196,39 +197,38 @@ public class SessionBackgroundTaskManager {
|
|
|
|
|
|
|
|
|
|
|
|
return true
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// Returns `false` if the background task cannot be begun.
|
|
|
|
/// Returns `false` if the background task cannot be begun
|
|
|
|
private func startBackgroundTask() -> Bool {
|
|
|
|
///
|
|
|
|
|
|
|
|
/// **Note:** Should only be called internally within `queue.sync` for thread safety
|
|
|
|
|
|
|
|
private func startOverarchingBackgroundTask() -> Bool {
|
|
|
|
guard dependencies[singleton: .appContext].isMainApp else { return false }
|
|
|
|
guard dependencies[singleton: .appContext].isMainApp else { return false }
|
|
|
|
|
|
|
|
|
|
|
|
return SessionBackgroundTaskManager.synced(self) { [weak self, dependencies] in
|
|
|
|
self.backgroundTaskId = dependencies[singleton: .appContext].beginBackgroundTask { [weak self] in
|
|
|
|
self?.backgroundTaskId = dependencies[singleton: .appContext].beginBackgroundTask {
|
|
|
|
|
|
|
|
/// Supposedly `[UIApplication beginBackgroundTaskWithExpirationHandler]`'s handler
|
|
|
|
/// Supposedly `[UIApplication beginBackgroundTaskWithExpirationHandler]`'s handler
|
|
|
|
/// will always be called on the main thread, but in practice we've observed otherwise.
|
|
|
|
/// will always be called on the main thread, but in practice we've observed otherwise.
|
|
|
|
///
|
|
|
|
///
|
|
|
|
/// See:
|
|
|
|
/// See:
|
|
|
|
/// https://developer.apple.com/documentation/uikit/uiapplication/1623031-beginbackgroundtaskwithexpiratio)
|
|
|
|
/// https://developer.apple.com/documentation/uikit/uiapplication/1623031-beginbackgroundtaskwithexpiratio)
|
|
|
|
|
|
|
|
self?.queue.sync {
|
|
|
|
self?.backgroundTaskExpired()
|
|
|
|
self?.backgroundTaskExpired()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// If the background task could not begin, return false to indicate that
|
|
|
|
// If the background task could not begin, return false to indicate that
|
|
|
|
return (self?.backgroundTaskId != .invalid)
|
|
|
|
return (self.backgroundTaskId != .invalid)
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// **Note:** Should only be called internally within `queue.sync` for thread safety
|
|
|
|
private func backgroundTaskExpired() {
|
|
|
|
private func backgroundTaskExpired() {
|
|
|
|
var backgroundTaskId: UIBackgroundTaskIdentifier = .invalid
|
|
|
|
let backgroundTaskId: UIBackgroundTaskIdentifier = self.backgroundTaskId
|
|
|
|
var expirationMap: [UInt64: () -> ()] = [:]
|
|
|
|
let expirationMap: [UInt64: () -> ()] = self.expirationMap
|
|
|
|
|
|
|
|
self.backgroundTaskId = .invalid
|
|
|
|
SessionBackgroundTaskManager.synced(self) { [weak self] in
|
|
|
|
self.expirationMap.removeAll()
|
|
|
|
backgroundTaskId = (self?.backgroundTaskId ?? .invalid)
|
|
|
|
|
|
|
|
self?.backgroundTaskId = .invalid
|
|
|
|
|
|
|
|
self?.expirationTimeObserver?.invalidate()
|
|
|
|
|
|
|
|
self?.expirationTimeObserver = nil
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
expirationMap = (self?.expirationMap ?? [:])
|
|
|
|
if self.expirationTimeObserver != nil {
|
|
|
|
self?.expirationMap.removeAll()
|
|
|
|
self.expirationTimeObserver?.cancel()
|
|
|
|
|
|
|
|
self.expirationTimeObserver = nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/// Supposedly `[UIApplication beginBackgroundTaskWithExpirationHandler]`'s handler
|
|
|
|
/// Supposedly `[UIApplication beginBackgroundTaskWithExpirationHandler]`'s handler
|
|
|
|
@ -247,33 +247,44 @@ public class SessionBackgroundTaskManager {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
private func continuityTimerDidFire() {
|
|
|
|
private func checkExpirationTime(in interval: DispatchTimeInterval) {
|
|
|
|
SessionBackgroundTaskManager.synced(self) { [weak self] in
|
|
|
|
expirationTimeObserver = DispatchSource.makeTimerSource(queue: queue)
|
|
|
|
self?.continuityTimer?.invalidate()
|
|
|
|
expirationTimeObserver?.schedule(deadline: .now() + interval)
|
|
|
|
self?.continuityTimer = nil
|
|
|
|
expirationTimeObserver?.setEventHandler { [weak self] in self?.expirationTimerDidFire() }
|
|
|
|
self?.ensureBackgroundTaskState()
|
|
|
|
expirationTimeObserver?.resume()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// Timer will always fire on the `queue` so no need to `queue.sync`
|
|
|
|
|
|
|
|
private func continuityTimerDidFire() {
|
|
|
|
|
|
|
|
continuityTimer = nil
|
|
|
|
|
|
|
|
ensureBackgroundTaskState()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// Timer will always fire on the `queue` so no need to `queue.sync`
|
|
|
|
private func expirationTimerDidFire() {
|
|
|
|
private func expirationTimerDidFire() {
|
|
|
|
|
|
|
|
expirationTimeObserver = nil
|
|
|
|
|
|
|
|
|
|
|
|
guard dependencies[singleton: .appContext].isMainApp else { return }
|
|
|
|
guard dependencies[singleton: .appContext].isMainApp else { return }
|
|
|
|
|
|
|
|
|
|
|
|
let backgroundTimeRemaining: TimeInterval = dependencies[singleton: .appContext].backgroundTimeRemaining
|
|
|
|
let backgroundTimeRemaining: TimeInterval = dependencies[singleton: .appContext].backgroundTimeRemaining
|
|
|
|
|
|
|
|
|
|
|
|
SessionBackgroundTaskManager.synced(self) { [weak self] in
|
|
|
|
/// It takes the OS a little while to update the 'backgroundTimeRemaining' value so if it hasn't been updated yet then don't do anything
|
|
|
|
// It takes the OS a little while to update the 'backgroundTimeRemaining' value so if it hasn't been updated
|
|
|
|
guard self.hasGottenValidBackgroundTimeRemaining == true || backgroundTimeRemaining != .greatestFiniteMagnitude else {
|
|
|
|
// yet then don't do anything
|
|
|
|
self.checkExpirationTime(in: .seconds(1))
|
|
|
|
guard self?.hasGottenValidBackgroundTimeRemaining == true || backgroundTimeRemaining != .greatestFiniteMagnitude else {
|
|
|
|
|
|
|
|
return
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
self?.hasGottenValidBackgroundTimeRemaining = true
|
|
|
|
self.hasGottenValidBackgroundTimeRemaining = true
|
|
|
|
|
|
|
|
|
|
|
|
// If there is more than 5 seconds remaining then no need to do anything yet (plenty of time to continue running)
|
|
|
|
switch backgroundTimeRemaining {
|
|
|
|
guard backgroundTimeRemaining <= 5 else { return }
|
|
|
|
/// There is more than 10 seconds remaining so no need to do anything yet (plenty of time to continue running)
|
|
|
|
|
|
|
|
case 10...: self.checkExpirationTime(in: .seconds(5))
|
|
|
|
|
|
|
|
|
|
|
|
// There isn't a lot of time remaining so trigger the expiration
|
|
|
|
/// There is between 5 and 10 seconds so poll more frequently just in case
|
|
|
|
self?.backgroundTaskExpired()
|
|
|
|
case 5..<10: self.checkExpirationTime(in: .milliseconds(2500))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// There isn't a lot of time remaining so trigger the expiration
|
|
|
|
|
|
|
|
default: self.backgroundTaskExpired()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
@ -282,8 +293,6 @@ public class SessionBackgroundTaskManager {
|
|
|
|
|
|
|
|
|
|
|
|
public class SessionBackgroundTask {
|
|
|
|
public class SessionBackgroundTask {
|
|
|
|
private let dependencies: Dependencies
|
|
|
|
private let dependencies: Dependencies
|
|
|
|
|
|
|
|
|
|
|
|
/// This property should only be accessed while synchronized on this instance
|
|
|
|
|
|
|
|
private var taskId: UInt64?
|
|
|
|
private var taskId: UInt64?
|
|
|
|
private let label: String
|
|
|
|
private let label: String
|
|
|
|
private var completion: ((SessionBackgroundTaskState) -> ())?
|
|
|
|
private var completion: ((SessionBackgroundTaskState) -> ())?
|
|
|
|
@ -308,86 +317,31 @@ public class SessionBackgroundTask {
|
|
|
|
|
|
|
|
|
|
|
|
// MARK: - Functions
|
|
|
|
// MARK: - Functions
|
|
|
|
|
|
|
|
|
|
|
|
@discardableResult private static func synced<T>(_ lock: Any, closure: () -> T) -> T {
|
|
|
|
|
|
|
|
objc_sync_enter(lock)
|
|
|
|
|
|
|
|
let result: T = closure()
|
|
|
|
|
|
|
|
objc_sync_exit(lock)
|
|
|
|
|
|
|
|
return result
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private func startBackgroundTask() {
|
|
|
|
private func startBackgroundTask() {
|
|
|
|
// Make a local copy of completion to ensure that it is called exactly once
|
|
|
|
taskId = dependencies[singleton: .backgroundTaskManager].addTask { [weak self] in
|
|
|
|
var completion: ((SessionBackgroundTaskState) -> ())?
|
|
|
|
self?.taskExpired()
|
|
|
|
|
|
|
|
|
|
|
|
self.taskId = dependencies[singleton: .backgroundTaskManager].addTask { [weak self] in
|
|
|
|
|
|
|
|
Threading.dispatchMainThreadSafe {
|
|
|
|
|
|
|
|
guard let strongSelf = self else { return }
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
SessionBackgroundTask.synced(strongSelf) {
|
|
|
|
|
|
|
|
self?.taskId = nil
|
|
|
|
|
|
|
|
completion = self?.completion
|
|
|
|
|
|
|
|
self?.completion = nil
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
completion?(.expired)
|
|
|
|
if taskId == nil {
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// If we didn't get a taskId then the background task could not be started so
|
|
|
|
|
|
|
|
// we should call the completion block with a 'couldNotStart' error
|
|
|
|
|
|
|
|
guard taskId == nil else { return }
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
SessionBackgroundTask.synced(self) { [weak self] in
|
|
|
|
|
|
|
|
completion = self?.completion
|
|
|
|
|
|
|
|
self?.completion = nil
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if completion != nil {
|
|
|
|
|
|
|
|
Threading.dispatchMainThreadSafe {
|
|
|
|
|
|
|
|
completion?(.couldNotStart)
|
|
|
|
completion?(.couldNotStart)
|
|
|
|
}
|
|
|
|
completion = nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
public func cancel() {
|
|
|
|
public func cancel() {
|
|
|
|
guard taskId != nil else { return }
|
|
|
|
guard taskId != nil else { return }
|
|
|
|
|
|
|
|
|
|
|
|
// Make a local copy of completion to ensure that it is called exactly once
|
|
|
|
dependencies[singleton: .backgroundTaskManager].removeTask(taskId: taskId)
|
|
|
|
var completion: ((SessionBackgroundTaskState) -> ())?
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
SessionBackgroundTask.synced(self) { [weak self, dependencies] in
|
|
|
|
|
|
|
|
dependencies[singleton: .backgroundTaskManager].removeTask(taskId: self?.taskId)
|
|
|
|
|
|
|
|
completion = self?.completion
|
|
|
|
|
|
|
|
self?.taskId = nil
|
|
|
|
|
|
|
|
self?.completion = nil
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// endBackgroundTask must be called on the main thread.
|
|
|
|
|
|
|
|
if completion != nil {
|
|
|
|
|
|
|
|
Threading.dispatchMainThreadSafe {
|
|
|
|
|
|
|
|
completion?(.cancelled)
|
|
|
|
completion?(.cancelled)
|
|
|
|
}
|
|
|
|
completion = nil
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
private func endBackgroundTask() {
|
|
|
|
private func endBackgroundTask() {
|
|
|
|
guard taskId != nil else { return }
|
|
|
|
cancel()
|
|
|
|
|
|
|
|
|
|
|
|
// Make a local copy of completion since this method is called by `dealloc`
|
|
|
|
|
|
|
|
var completion: ((SessionBackgroundTaskState) -> ())?
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
SessionBackgroundTask.synced(self) { [weak self, dependencies] in
|
|
|
|
|
|
|
|
dependencies[singleton: .backgroundTaskManager].removeTask(taskId: self?.taskId)
|
|
|
|
|
|
|
|
completion = self?.completion
|
|
|
|
|
|
|
|
self?.taskId = nil
|
|
|
|
|
|
|
|
self?.completion = nil
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// endBackgroundTask must be called on the main thread.
|
|
|
|
private func taskExpired() {
|
|
|
|
if completion != nil {
|
|
|
|
completion?(.expired)
|
|
|
|
Threading.dispatchMainThreadSafe {
|
|
|
|
completion = nil
|
|
|
|
completion?(.cancelled)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
Morgan Pretty
3 weeks ago
committed by
GitHub