Doobie. Compose .update.withGeneratedKeys() and .update.run

Question

参考这个问题。
我想通过某种条件插入一些实体。它可以插入，也可以不插入。如果条件为 true，则插入实体。我想在各种表中插入一些其他数据。它看起来像这样：

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....应用可以编译，但第二行中的inserts不会发生。

---

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，则可以使用以下事实：如果生成的Streamq.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