update task cache when bundle is deleted

Author: CollinBeczak

app/org/maproulette/framework/repository/TaskBundleRepository.scala

@@ -10,7 +10,7 @@ import org.slf4j.LoggerFactory
 import anorm.ToParameterValue
 import anorm._, postgresql._
 import javax.inject.{Inject, Singleton}
-import org.maproulette.exception.{InvalidException}
+import org.maproulette.exception.{InvalidException, NotFoundException}
 import org.maproulette.Config
 import org.maproulette.framework.psql.Query
 import org.maproulette.framework.psql.filter.BaseParameter
@@ -33,6 +33,7 @@ class TaskBundleRepository @Inject() (
     with Locking[Task] {
   protected val logger           = LoggerFactory.getLogger(this.getClass)
   implicit val baseTable: String = Task.TABLE
+  val cacheManager               = this.taskRepository.cacheManager
 
   /**
     * Inserts a new task bundle with the given tasks, assigning ownership of
@@ -144,8 +145,15 @@ class TaskBundleRepository @Inject() (
   def deleteTaskBundle(user: User, bundle: TaskBundle, primaryTaskId: Option[Long] = None): Unit = {
     this.withMRConnection { implicit c =>
       SQL(
-        "UPDATE tasks SET bundle_id = NULL, is_bundle_primary = NULL WHERE bundle_id = {bundleId}"
-      ).on(Symbol("bundleId") -> bundle.bundleId).executeUpdate()
+        """UPDATE tasks 
+     SET bundle_id = NULL, 
+         is_bundle_primary = NULL 
+     WHERE bundle_id = {bundleId} OR id = {primaryTaskId}"""
+      ).on(
+          Symbol("bundleId")      -> bundle.bundleId,
+          Symbol("primaryTaskId") -> primaryTaskId
+        )
+        .executeUpdate()
 
       if (primaryTaskId != None) {
         // unlock tasks (everything but the primary task id)
@@ -164,6 +172,23 @@ class TaskBundleRepository @Inject() (
         }
       }
 
+      // Update cache for each task in the bundle
+      bundle.tasks match {
+        case Some(t) =>
+          for (task <- t) {
+            this.cacheManager.withOptionCaching { () =>
+              Some(
+                task.copy(
+                  bundleId = None,
+                  isBundlePrimary = None
+                )
+              )
+            }
+          }
+
+        case None => // no tasks in bundle
+      }
+
       // Delete from task_bundles which will also cascade delete from bundles
       SQL("DELETE FROM task_bundles WHERE bundle_id = {bundleId}")
         .on(Symbol("bundleId") -> bundle.bundleId)