app/housekeeping/utils/BakesRepo.scala

package housekeeping.utils

import data.{Bakes, Dynamo}
import models.{Bake, BakeId, BakeStatus, NotificationConfig}

import scala.concurrent.ExecutionContext

// Wrapper around dynamo access.
// Injecting this functionality as a class facilitates unit testing.
class BakesRepo(notificationConfig: Option[NotificationConfig])(implicit
    dynamo: Dynamo,
    exec: ExecutionContext
) {

  def getBakes: List[Bake.DbModel] = Bakes.scanForAll()

  def updateStatusToTimedOutIfRunning(bakeId: BakeId): Unit = {
    // Couldn't work out how to specify that the instance should be running in the update request.
    // Therefore, look up the bake first and check its status.
    Bakes.findById(bakeId.recipeId, bakeId.buildNumber).foreach { bake =>
      if (bake.status == BakeStatus.Running) {
        Bake.updateStatusAndNotifyFailure(
          bakeId,
          BakeStatus.TimedOut,
          notificationConfig
        )
      }
    }
  }
}