phatmann · April 15, 2015 03:14 · Apr 15, 2015
diff --git a/BackgroundTask.swift b/BackgroundTask.swift
@@ -0,0 +1,30 @@
+class BackgroundTask {
+    private let application: UIApplication
+    private var identifier = UIBackgroundTaskInvalid
+
+    init(application: UIApplication) {
+        self.application = application
+    }
+
+    class func run(application: UIApplication, handler: (BackgroundTask) -> ()) {
+        // NOTE: The handler must call end() when it is done
+
+        let backgroundTask = BackgroundTask(application: application)
+        backgroundTask.begin()
+        handler(backgroundTask)
+    }
+
+    func begin() {
+        self.identifier = application.beginBackgroundTaskWithExpirationHandler {
+            self.end()
+        }
+    }
+
+    func end() {
+        if (identifier != UIBackgroundTaskInvalid) {
+            application.endBackgroundTask(identifier)
+        }
+
+        identifier = UIBackgroundTaskInvalid
+    }
+}