Выполнение действий, не связанных с базой данных, в транзакции в Slick 3


У меня возникли проблемы с пониманием нового API Slick DBIOAction, который, похоже, не имеет большого количества примеров в документах. Я использую Slick 3.0.0, и мне нужно выполнить некоторые действия БД, а также некоторые вычисления с данными, полученными из базы данных, но все эти действия должны быть выполнены внутри одной транзакции. Я пытаюсь сделать следующее:

  1. выполнить запрос к базе данных (таблица types).
  2. выполните некоторые агрегации и фильтрацию результатов запроса. (этот расчет не может быть выполнен в базе данных).
  3. выполните другой запрос, основанный на вычислениях из шага 2 (Таблица messages - из-за некоторых ограничений этот запрос должен быть в raw SQL).
  4. соедините данные из шага 2 и 3 в памяти.
Я хочу, чтобы запросы из шагов 1 и 3 выполнялись внутри транзакции, так как данные из их результирующих наборов должны быть согласованными.

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

  val compositeAction = (for {
    rawTypes <- TableQuery[DBType].result
    (projectId, types) <- rawTypes.groupBy(_.projectId).toSeq.map(group => (group._1, group._2.slice(0, 10)))
    counts <- DBIO.sequence(types.map(aType => sql"""select count(*) from messages where type_id = ${aType.id}""".as[Int]))
  } yield (projectId, types.zip(counts))).transactionally
    Первая строка понимания for выбирает данные из таблицы types.
  1. второй ряд for понимания, как предполагается, делает некоторую группировку и нарезку результатов, в результате чего Seq[(Option[String], Seq[String])]
  2. Третья строка понимания for должна выполнить набор запросов для каждого элемента из предыдущего шага, в частности, она должна выполнить один SQL-запрос для каждого из значений внутри Seq[String]. Так в третьем ряду я строю последовательность DBIOActions.
  3. предложение yield zips types из второго шага и counts из третьего шага.

Эта конструкция, однако, не работает и дает две ошибки времени компиляции:

Error:(129, 16) type mismatch;
 found   : slick.dbio.DBIOAction[(Option[String], Seq[(com.centreit.proto.repiso.storage.db.models.DBType#TableElementType, Vector[Int])]),slick.dbio.NoStream,slick.dbio.Effect]
    (which expands to)  slick.dbio.DBIOAction[(Option[String], Seq[(com.centreit.proto.repiso.storage.db.models.TypeModel, Vector[Int])]),slick.dbio.NoStream,slick.dbio.Effect]
 required: scala.collection.GenTraversableOnce[?]
        counts <- DBIO.sequence(types.map(aType => sql"""select count(*) from messages where type_id = ${aType.id}""".as[Int]))
               ^
Error:(128, 28) type mismatch;
 found   : Seq[Nothing]
 required: slick.dbio.DBIOAction[?,?,?]
        (projectId, types) <- rawTypes.groupBy(_.projectId).toSeq.map(group => (group._1, group._2.slice(0, 10)))
                           ^

Я попытался обернуть вторую строку в DBIOAction с помощью DBIO.successful, который должен поднять постоянное значение в DBIOAction монаду:

(projectId, types) <- DBIO.successful(rawTypes.groupBy(_.projectId).toSeq.map(group => (group._1, group._2.slice(0, 10))))

Но в этом коде переменная types выводится как Any, и код этого не делает. компиляция из-за этого.

1 24

1 ответ:

Попробуйте сделать это следующим образом:

val compositeAction = (for {
  rawTypes <- TableQuery[DBType].result
  pair <- DBIO.sequence(rawTypes.groupBy(_.projectId).toSeq.map(group => DBIO.successful(group)))
  counts <- DBIO.sequence(pair.head._2.map(aType => sql"""select count(*) from messages where type_id = ${aType.id}""".as[Int]))
} yield (pair.head._1, pair.head._2.zip(counts))).transactionally