Create `JobTracker` helper class

1 files changed, 46 insertions, 0 deletions

diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/util/JobTracker.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/util/JobTracker.kt
new file mode 100644
index 0000000000..4934669d58
--- /dev/null
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/util/JobTracker.kt
@@ -0,0 +1,46 @@
+package net.mullvad.mullvadvpn.util
+
+import kotlinx.coroutines.launch
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.GlobalScope
+import kotlinx.coroutines.Job
+
+class JobTracker {
+    private val jobs = HashMap<Long, Job>()
+
+    private var jobIdCounter = 0L
+
+    fun newJob(job: Job): Long {
+        synchronized(jobs) {
+            val jobId = jobIdCounter
+
+            jobIdCounter += 1
+
+            jobs.put(jobId, GlobalScope.launch(Dispatchers.Default) {
+                job.join()
+
+                synchronized(jobs) {
+                    jobs.remove(jobId)
+                }
+            })
+
+            return jobId
+        }
+    }
+
+    fun cancelJob(jobId: Long) {
+        synchronized(jobs) {
+            jobs.remove(jobId)?.cancel()
+        }
+    }
+
+    fun cancelAllJobs() {
+        synchronized(jobs) {
+            for (job in jobs.values) {
+                job.cancel()
+            }
+
+            jobs.clear()
+        }
+    }
+}