Slick. Выберите и затем обновите выбранные строки.

Question

Я хочу выбрать некоторые строки, обновить их и вернуть обновленные значения, но я не понимаю, как это сделать с помощью Slick. Вот пример. Я хочу, чтобы выбрать все задачи, которые ожидающие выполнения, блокировать их, изменение статусов в ходе и вернуться обновленные задачи:

object Test { 

    case class Task(id: Int, status: String) 

    class TaskTable(tag: Tag) extends Table[Task](tag, "tasks") { 
    def id = column[Int]("id") 
    def status = column[String]("status") 

    def * = (id, status) <>(Task.tupled, Task.unapply) 
    } 

    val tasks = TableQuery[TaskTable] 

    def selectWaitingTasksAndChangeStatus(): Seq[Task] = { 
    tasks.filter(_.status === "awaitingExecution").forUpdate 
    // Here I want to change status to "inProgress" and 
    // return tasks to client code with "inProgress" status 
    } 

} 

Answer 1

ли это, что вы ищете?

import scala.concurrent.Await 
import scala.concurrent.duration.Duration 

def selectWaitingTasksAndChangeStatus(): Seq[Task] = { 
    val selectAction = tasks.filter(_.status === "inProgress").result 
    val updateAction = tasks.filter(_.status === "awaitingExecution").map(_.status).update("inProgress") 
    val combinedAction = for { 
    tasksBeforeUpdate <- selectAction 
    _     <- updateAction 
    tasksAfterUpdate <- selectAction 
    } yield tasksAfterUpdate.diff(tasksBeforeUpdate) 

    Await.result(db.run(combinedAction.transactionally), Duration.Inf) 
} 

Так как вы хотели бы получить Seq[Task] от этого метода, вы должны синхронно ждать результата из базы данных. Для асинхронного решения в качестве возвращаемого типа потребуется Future[Seq[Task]].