Use hashmap on AndroidTaskRunner

TheMaxul · Jul 29, 2016 · 3a7f277 · 3a7f277
1 parent 3a3be66
commit 3a7f277
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/app/src/main/java/org/isoron/uhabits/tasks/AndroidTaskRunner.java b/app/src/main/java/org/isoron/uhabits/tasks/AndroidTaskRunner.java
@@ -33,9 +33,12 @@ public class AndroidTaskRunner implements TaskRunner
 {
     private final LinkedList<CustomAsyncTask> activeTasks;
 
+    private final HashMap<Task, CustomAsyncTask> taskToAsyncTask;
+
     public AndroidTaskRunner()
     {
         activeTasks = new LinkedList<>();
+        taskToAsyncTask = new HashMap<>();
     }
 
     @Provides
@@ -54,8 +57,9 @@ public void execute(Task task)
     @Override
     public void publishProgress(Task task, int progress)
     {
-        for (CustomAsyncTask asyncTask : activeTasks)
-            if (asyncTask.getTask() == task) asyncTask.publish(progress);
+        CustomAsyncTask asyncTask = taskToAsyncTask.get(task);
+        if(asyncTask == null) return;
+        asyncTask.publish(progress);
     }
 
     private class CustomAsyncTask extends AsyncTask<Void, Integer, Void>
@@ -89,12 +93,14 @@ protected void onPostExecute(Void aVoid)
         {
             task.onPostExecute();
             activeTasks.remove(this);
+            taskToAsyncTask.remove(task);
         }
 
         @Override
         protected void onPreExecute()
         {
             activeTasks.add(this);
+            taskToAsyncTask.put(task, this);
             task.onPreExecute();
         }