在Scala中Doobie上的事务内获取或插入

import doobie._
import doobie.implicits._
import cats.effect.{ContextShift, IO}
import doobie.util.transactor.Transactor

val xa = Transactor.fromDriverManager[IO](
  "org.postgresql.Driver",     // 数据库驱动
  "jdbc:postgresql:database",  // 数据库连接URL
  "username",                  // 数据库用户名
  "password"                   // 数据库密码
)

def insertData(data: Data): ConnectionIO[Int] = {
  sql"INSERT INTO table (column1, column2) VALUES (${data.field1}, ${data.field2})"
    .update
    .run
}

val result: IO[Int] = (for {
  _ <- sql"BEGIN".update.run.transact(xa)  // 开始事务
  count <- insertData(data).transact(xa)   // 执行数据库操作
  _ <- sql"COMMIT".update.run.transact(xa) // 提交事务
} yield count).handleErrorWith { e =>
  sql"ROLLBACK".update.run.transact(xa).flatMap(_ => IO.raiseError(e)) // 回滚事务并抛出异常
}