如何用Doobie正确处理Hikari连接池

Question

我一直在使用 doobie (cats) 从 scalatra 应用程序连接到 postgresql 数据库。最近我注意到该应用程序正在为每个事务创建一个新的连接池。我最终解决了它 - 见下文，但这种方法与 'managing connections' section of the book of doobie 中采用的方法完全不同，我希望有人可以确认它是否明智或者是否有更好的方法来设置连接池。

这是我最初拥有的 - 这有效但会在每个连接上创建一个新的连接池：

import com.zaxxer.hikari.HikariDataSource
import doobie.hikari.hikaritransactor.HikariTransactor
import doobie.imports._

val pgTransactor = HikariTransactor[IOLite](
  "org.postgresql.Driver",
  s"jdbc:postgresql://${postgresDBHost}:${postgresDBPort}/${postgresDBName}",
  postgresDBUser,
  postgresDBPassword
)
// every query goes via this function
def doTransaction[A](update: ConnectionIO[A]): Option[A] = {
    val io = for {
      xa <- pgTransactor
      res <- update.transact(xa) ensuring xa.shutdown
    } yield res
    io.unsafePerformIO
}

我最初的假设是问题是每个请求都有 ensuring xa.shutdown，但删除它会导致连接很快用完，直到剩下 none。

这是解决问题的尝试 - 使我能够删除 ensuring xa.shutdown，但仍然导致连接池被反复打开和关闭：

val pgTransactor: HikariTransactor[IOLite] = HikariTransactor[IOLite](
  "org.postgresql.Driver",
  s"jdbc:postgresql://${postgresDBHost}:${postgresDBPort}/${postgresDBName}",
  postgresDBUser,
  postgresDBPassword
).unsafePerformIO

def doTransaction[A](update: ConnectionIO[A]): Option[A] = {
    val io = update.transact(pgTransactor)
    io.unsafePerformIO
}

最后，我通过创建 HikariDataSource 对象并将其传递给 HikariTransactor 构造函数来获得所需的行为：

val dataSource = new HikariDataSource()
dataSource.setJdbcUrl(s"jdbc:postgresql://${postgresDBHost}:${postgresDBPort}/${postgresDBName}")
dataSource.setUsername(postgresDBUser)
dataSource.setPassword(postgresDBPassword)

val pgTransactor: HikariTransactor[IOLite] = HikariTransactor[IOLite](dataSource)

def doTransaction[A](update: ConnectionIO[A], operationDescription: String): Option[A] = {
  val io = update.transact(pgTransactor)
  io.unsafePerformIO
}

Answer 1

你可以这样做：

val xa = HikariTransactor[IOLite](dataSource).unsafePerformIO

并将其传递到您的存储库。 .transact 应用事务边界，如 Slick 的 .transactionally。例如：

def interactWithDb = {
  val q: ConnectionIO[Int] = sql"""..."""
  q.transact(xa).unsafePerformIO
}

Answer 2

是的，Radu 的回复解决了问题。 HikariTransactor（实际上是底层 HikariDataSource）具有内部状态，因此构建它是一个副作用；并且您想在程序启动时执行一次并根据需要传递它。所以您的解决方案有效，只需注意副作用。

此外，如前所述，我不监控 SO ……如果您有疑问，请尝试 Gitter 频道或打开问题。 :-)

如何用Doobie正确处理Hikari连接池

How to correctly handle Hikari connection pool with Doobie

postgresql

scala

connection-pooling

hikaricp

doobie