首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

Slick 数据库连接和事务处理

数据库连接和事务处理

你可以在程序的任何地方使用数据库查询,当执行查询时你需要有一个数据库连接。

你可以通过创建一个 Database 对象来连接一个 JDBC 数据库,有多种方法可以创建一个数据库对象。

使用 JDBC URL

你可以使用 JDBC URL 来创建一个 Database 对象(URL 的格式取决于连接的数据库的类型),比如:

代码语言:txt
复制
val db = Database.forURL("jdbc:h2:mem:test1;DB_CLOSE_DELAY=-1", driver="org.h2.Driver")

创建一个基于内存的 H2 数据库连接,再比如我们之前使用的 MySQL 数据库,可以使用

代码语言:txt
复制
val db = Database.forURL("jdbc:mysql://127.0.0.1/Chinook",
    driver = "com.mysql.jdbc.Driver",
    user="user",
    password="password")

使用 DataSource

你可以使用已有的 datasource 对象,来构建一个 Database 对象,比如你从连接池中取得一个 Datasource 对象,然后连接到 Slick 库中

代码语言:txt
复制
val db = Database.forDataSource(dataSource: javax.sql.DataSource)

之后你创建一个 Session 对象,将从连接池中取得一个数据库连接,当关闭 Session 时,连接退回给连接池以作他用。

使用 JNDI 名称

如果你使用 JNDI,你可以提供 JNDI 名称来构建一个 Database 对象:

代码语言:txt
复制
val db = Database.forName(jndiName: String)

Session 管理

现在你有了一个数据库对象可以打开一个数据库(Slick 函数库封装了一个 Session 对象)

Database 的 withSession 方法,创建一个 Session 对象,它可以传递给一个函数,函数返回时自动关闭这个 Session 对象,如果你使用连接池,关闭 Session 对象,自动将连接退回连接池。

代码语言:txt
复制
val query = for (c <- coffees) yield c.name
val result = db.withSession {
    session =>
    query.list()( session )
}

你可以看到,我们可以在 withSession 之外定义查询,只有在实际执行查询时才需要一个 Session 对象,要注意的是 Session 的缺省模式为自动提交(auto-commit )模式。每个数据库指令(比如 insert )都自动提交给数据库。 如果需要将几个指令作为一个整体,那么就需要使用事务处理(Transaction)上面的例子,我们在执行查询时,明确指明了 session 对象,你可以使用隐含对象来避免这种情况,比如:

代码语言:txt
复制
val query = for (c <- coffees) yield c.name
val result = db.withSession {
    implicit session =>
    query.list // <- takes session implicitly
}
// query.list // <- would not compile, no implicit value of type Session

手工管理 Session

这不是推荐使用的情况,但如果你需要自己管理 Session 对象,你可以自己管理 Session 的生命周期:

代码语言:txt
复制
val query = for (c <- coffees) yield c.name
val session : Session = db.createSession
val result  = query.list()( session )
session.close

事务处理

你可以使用 Session 对象的 withTransaction 方法来创建一个事务,传给该方法的语句将作为一个整体事务执行,如果出现异常,Slick 自动回滚事务,你也可以使用 rollback 强制事务回滚,要注意的是 Slick 只会回滚数据库相关操作,而不会取消其它 Scala 语句。

代码语言:txt
复制
session.withTransaction {
    // your queries go here

    if (/* some failure */ false){
        session.rollback // signals Slick to rollback later
    }

} //

如果你没有 Session 对象,也可以直接使用数据库对象的 withTransaction 方法,如:

代码语言:txt
复制
db.withTransaction{
    implicit session =>
    // your queries go here
}

扫码关注腾讯云开发者

领取腾讯云代金券