Выполнение действий, не связанных с базой данных, в транзакции в Slick 3
У меня возникли проблемы с пониманием нового API Slick DBIOAction
, который, похоже, не имеет большого количества примеров в документах. Я использую Slick 3.0.0, и мне нужно выполнить некоторые действия БД, а также некоторые вычисления с данными, полученными из базы данных, но все эти действия должны быть выполнены внутри одной транзакции. Я пытаюсь сделать следующее:
- выполнить запрос к базе данных (таблица
types
). - выполните некоторые агрегации и фильтрацию результатов запроса. (этот расчет не может быть выполнен в базе данных).
- выполните другой запрос, основанный на вычислениях из шага 2 (Таблица
messages
- из-за некоторых ограничений этот запрос должен быть в raw SQL). - соедините данные из шага 2 и 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
понимания, как предполагается, делает некоторую группировку и нарезку результатов, в результате чегоSeq[(Option[String], Seq[String])]
Третья строка понимания - предложение
yield
zip
stypes
из второго шага иcounts
из третьего шага.
for
выбирает данные из таблицы types
.
for
должна выполнить набор запросов для каждого элемента из предыдущего шага, в частности, она должна выполнить один SQL-запрос для каждого из значений внутри Seq[String]
. Так в третьем ряду я строю последовательность DBIOAction
s.
Эта конструкция, однако, не работает и дает две ошибки времени компиляции:
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 ответ:
Попробуйте сделать это следующим образом:
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