Doobie. Создайте.update.withGeneratedKeys() и.update.run

Question

Ссылка на этот вопрос.
Я хочу вставить некоторую сущность по какому-то условию. Это может быть вставлено или нет. Тогда условие истинно, и сущность вставлена, я хочу вставить некоторые другие данные в различные таблицы. Это выглядит так:

val q = sql"insert into some_table (some_field) select 42 where ...(some condition)"

val inserts = List(
  sql"insert ...",
  sql"insert ...",
  sql"insert ..."
)

for {
  id <- q.update.withGeneratedKeys[Long]("id")   
  _ <- inserts.reduce(_ ++ _).update.run
} yield id

Проблема в том, что это не компилируется, потому что первая вставка fs2.Stream но второй нет.

Меня пытались заменить _ <- inserts.reduce... в _ = inserts.reduce, Приложение становится компилируемым, но вставок во второй строке не происходит.

---

UPD
Мой возможный способ решить эту проблему:

...
for {
  idOpt <- q.update.withGeneratedKeys[Long]("id").compile.last   
  _ <- idOpt.fold(0.pure[ConnectionIO])(_ => inserts.reduce(_ ++ _).update.run)
} yield idOpt

Это работает, но ИМХО это не красиво. Есть ли способ сделать это лучше?

Answer 1

Один из способов выполнения пакетных вставок - если у вас есть похожие данные - это использовать updateMany- см. документ:

import doobie._
type PersonInfo = (String, Option[Short])

def insertMany(ps: List[PersonInfo]): ConnectionIO[Int] = {
  val sql = "insert into person (name, age) values (?, ?)"
  Update[PersonInfo](sql).updateMany(ps)
}

// Some rows to insert
val data = List[PersonInfo](
  ("Frank", Some(12)),
  ("Daddy", None))

Также, если вы удалите.compile.last, вы можете использовать тот факт, что если ваш результат Stream q.update.withGeneratedKeys[Long]("id") является empty, вы выходите раньше for-comprehension.

Итак, вот что вы могли бы сделать:

import fs2.Stream

val result =
  // Now the for-comprehension operates on a Stream instead of an Option
  for {
    r <- q.update.withGeneratedKeys[Long]("id")   
    _ <- Stream.eval(insertMany(data)) // insertMany(data) is defined like in the snippet above
  } yield r

result.compile.last