如何使用 Scala 的 Slick 库执行 JDBC 身份验证？

Question

介绍

• 我有一个用 Akka HTTP 编写的微服务，它通过 REST ZDB974238714CA8ADE4F634A 与前端通信。
• 我使用 Scala 的 Slick 库与 PostgreSQL DBMS 进行通信。
• 我想使用 HTTP 的基本身份验证机制保护 API（前端在每个请求中向微服务发送用户名和密码）。
• 从每个 HTTP API 调用中提取凭据我想添加到 Slick 的操作中，以确保通过“JDBC 身份验证”确保用户有权操作给定的数据集。

问题

虽然对我来说很明显如何使用普通 Java 的 DriverManager 通过使用一组凭据创建每个新连接并使用此连接来执行未来的操作来完成此操作，但 Slick 库提供了如下内容：

val db = Database.forURL(jdbcUrl, superuser, password)

但文档指出，在应用程序的整个生命周期中只应创建一个这样的实例：

注意：一个数据库 object 通常管理一个线程池和一个连接池。 当不再需要它时，您应该始终正确关闭它（除非 JVM 进程无论如何都会终止）。 不要为每个数据库操作创建一个新数据库。 单个实例旨在在您的应用程序的整个生命周期内保持活动状态。

所以我坚持使用一个db singleton，它始终使用超级用户凭据运行所有操作。

我是 Slick 的新手，我不知道如何添加额外的用户名和密码，以便将其与操作（SQL、DDL 语句）一起发送到可以验证用户是否被允许执行此操作的 DBMS。 这是我想要实现的示例：

  override def removeDatabase(name: String): Future[Unit] =
    db.run(sqlu"""DROP DATABASE IF EXISTS #${name.toLowerCase};""").map(_ => ())

对于此删除操作，我想添加从 HTTP 请求中提取的凭据，以获得 JDBC 身份验证，以确保执行此操作的当前用户有权删除此数据库。

到目前为止我做了什么？

根据 Slick 文档：

为了将 Slick 中不可用的功能降至 JDBC 级别，您可以使用 SimpleDBIO 操作，该操作在数据库线程上运行并访问 JDBC 连接： val getAutoCommit = SimpleDBIO[Boolean](_.connection.getAutoCommit)

我试过这个：

override def removeDatabase(name: String): Future[Unit] = {
  val dropStatement = SimpleDBIO[Unit] { session => {
    val currentConnection = session.connection
    currentConnection.setClientInfo("username", "testUsername")
    currentConnection.setClientInfo("password", "invalidPwd")
    println(">>> username: " + currentConnection.getClientInfo("username"))
    println(">>> password: " + currentConnection.getClientInfo("password"))
    sqlu"""DROP DATABASE IF EXISTS #${name.toLowerCase};"""
  }
  }
  db.run(dropStatement).map(_ => ())
}

但不幸的是，它没有任何影响，传递给客户端信息的凭据被忽略：

>>> username: null 
>>> password: null

Answer 1

您可以像这样创建自定义Database ：

import slick.jdbc.PostgresProfile.api._

val db: Database = Database.forConfig("myDatabase")

这使用来自application.conf的适当配置：

myDatabase = {
  dataSourceClass = "org.postgresql.ds.PGSimpleDataSource"
  properties = {
    databaseName = "mydata"
    user = "postgres"
    password = "postgres"
  }
  numThreads = 4
}