scala - 如何使用 doobie 对 Scala 中的 PostgreSQL 数据库执行字符串 SQL 语句列表?
问题描述
我将以下 10 行 Python 代码移植到 Scala:
import psycopg2
def execute(user, password, database, host, port, *queries):
connection = psycopg2.connect(user=user, password=password, host=host, port=port, database=database)
cursor = connection.cursor()
for sql in queries:
print(sql)
cursor.execute(sql)
connection.commit()
cursor.close()
connection.close()
我有以下等效的 Scala 代码:
def execute(user: String, password: String, database: String, host: String, port: Int, queries: String*): Unit = {
???
}
我想在单个事务中针对数据库执行(并打印)一堆 SQL 语句(假设它是 Postgres)并完成。
我如何使用doobie做到这一点?
笔记:
我无法将接口更改为我的
execute()
(包括我无法添加类型或隐式参数)。它必须接受字符串用户、密码等和一个可变参数,queries: String*
从而保持接口与 Python 接口相同。还请提及所需的所有进口
解决方案
您可以使用 for-comprehension 在 doobie 的一个事务中运行多个查询,例如:
val query = for {
_ <- sql"insert into person (name, age) values ($name, $age)".update.run
id <- sql"select lastval()".query[Long].unique
} yield p
但是这个解决方案不适用于你的情况,因为你有一个动态的查询列表。幸运的是,我们可以使用猫的遍历:
import cats.effect.ContextShift
import doobie._
import doobie.implicits._
import cats.effect._
import scala.concurrent.ExecutionContext
import cats.implicits._
import cats._
import cats.data._
def execute(user: String, password: String, database: String, host: String, port: Int, queries: String*): Unit = {
//you can use other executor if you want
//it would be better to pass context shift as implicit argument to method
implicit val cs: ContextShift[IO] = IO.contextShift(ExecutionContext.global)
//let's create transactor
val xa = Transactor.fromDriverManager[IO](
"org.postgresql.Driver",
s"jdbc:postgresql://$host:$port/$database", //remember to change url or make it dynamic, if you run it agains another database
user,
password
)
val batch = queries
.toList //we need to change String* to list, since String* doesn't have necessary typeclass for Aplicative
.traverse(query => Update0(query, None).run) //we lift strings to Query0 and then run them, then we change List[ConnectionIO[Int]] to ConnectionIO[List[Int]]
//above can be done in two steps using map and sequence
batch //now we've got single ConnectionIO which will run in one transaction
.transact(xa) //let's make it IO[Int]
.unsafeRunSync() //we need to block since your method returns Unit
}
可能您的 IDE 会显示此代码无效,但它是正确的。IDE 无法处理 Scala 魔法。
您也可以考虑使用unsafeRunTimed
而不是unsafeRunSync
添加时间限制。
另外,请记住将用于 jdbc和猫的postgresql 驱动程序添加到您的. Doobie 在后台使用猫,但我认为显式依赖可能是必要的。build.sbt
推荐阅读
- xslt - xslt 获取祖先文件夹路径
- sql - 在 SQL 选择查询中使用并集排序子句
- apache-spark - 如何规范化火花数据框的多列
- laravel - 未能加载外部实体“https://ws.hipay.com/soap/payment-v2?wsdl”
- python - 如何在create()odoo 10中复制字段的值
- javascript - 点击标签a时如何向href url添加参数?
- android - 如何通过 id 更改搜索位置?
- html - 如何在 Booststrap 4 中集中导航选项卡/药丸?
- laravel - 模型不显示数据,但保存数据,Laravel
- jquery - 使用日期选择器控件停止用户在输入中输入日期