Подтвердить что ты не робот

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

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

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

4b9b3361

Ответ 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