Scala数据库操作

Scala各种数据库操作的框架已经出来了，例如：ScalaQuery、O/R
Broker 、Squeryl 等。但是，我看过的Scala编程书从未有介绍数据库操作的章节，于是就很好奇，查了一些资料，最终自己写了一个简单的基于Scala的数据库操作代码。

1.连接数据库

和Java没有太大的不同，还是定义数据库四个几个参数，操作步骤和JDBC一样样儿的。唯一需要注意的一点是：JDBC中Class.forName()换成了classOf[com.mysql.jdbc.Driver]，并且“com.mysql.jdbc.Driver”是不能加引号的。

闲言少叙，下面直接看代码吧。

package edu.wzm.db.scala

import java.sql.Connection
import java.sql.DriverManager
//import java.sql.PreparedStatement
//import java.sql.ResultSet

/**
* Created by GatsbyNewton on 2015/10/12.
*/
object DBUtils {

val url = "jdbc:mysql://localhost:3306/test"
val username = "root"
val password = "admin"

classOf[com.mysql.jdbc.Driver]

def getConnection(): Connection = {
DriverManager.getConnection(url, username, password)
}

def close(conn: Connection): Unit = {
try{
if(!conn.isClosed() || conn != null){
conn.close()
}
}
catch {
case ex: Exception => {
ex.printStackTrace()
}
}
}

//  def close(rs: ResultSet): Unit = {
//    try {
//      if(!rs.isClosed() || rs != null){
//        rs.close()
//      }
//    }
//    catch {
//      case ex: Exception => {
//        ex.printStackTrace()
//      }
//    }
//  }
//
//  def close(pstm: PreparedStatement): Unit = {
//    try {
//      if(!pstm.isClosed() || pstm != null){
//        pstm.close()
//      }
//    }
//    catch {
//      case ex: Exception => {
//        ex.printStackTrace()
//      }
//    }
//  }

}

看完代码后，可能有小伙伴会问“为啥把ResultSet和PreparedStatement的close代码注释掉”，在下面ACID的操作中就会发现原因。

2.ACID操作

package edu.wzm.db.scala

import scala.collection.mutable
import scala.collection.mutable.ArrayBuffer

/**
* Created by GatsbyNewton on 2015/10/12.
*/
class Operators {

case class User(id: String, name: String, age: Int)

//Insert operator
def add(user: User): Boolean = {
val conn = DBUtils.getConnection()
try{
val sql = new StringBuilder()
.append("INSERT INTO user(id, name, age)")
.append("     VALUES(?, ?, ?)")
val pstm = conn.prepareStatement(sql.toString())
pstm.setObject(1, user.id)
pstm.setObject(2, user.name)
pstm.setObject(3, user.age)

pstm.executeUpdate() > 0
}
finally {
conn.close()
}
}

//Delete operator
def delete(id: String): Boolean ={
val conn = DBUtils.getConnection()
try{
val sql = "DELETE FROM user WHERE id = ?"
val pstm = conn.prepareStatement(sql)
pstm.setObject(1, id)

pstm.executeUpdate() > 0
}
finally {
DBUtils.close(conn)
}
}

//Update operator
def modify(user: User): Boolean ={
val conn = DBUtils.getConnection()
try{
val sql = "UPDATE user SET age = ? WHERE id = ?"
val pstm = conn.prepareStatement(sql)
pstm.setObject(1, user.age)
pstm.setObject(2, user.id)

pstm.executeUpdate() > 0
}
finally {
DBUtils.close(conn)
}
}

//Select operator
def query(id: Int): ArrayBuffer[mutable.HashMap[String, Any]] = {
val conn = DBUtils.getConnection()
try{
val sql = new StringBuilder()
.append("SELECT name, age")
.append("  FROM user")
.append(" WHERE id >  ?")
val pstm = conn.prepareStatement(sql.toString())
pstm.setObject(1, id)
val rs = pstm.executeQuery()

val rsmd = rs.getMetaData()
val size = rsmd.getColumnCount()
val buffer = new ArrayBuffer[mutable.HashMap[String, Any]]()
while (rs.next()){
val map = mutable.HashMap[String, Any]()
for(i <- 1 to size) {
map += (rsmd.getColumnLabel(i) -> rs.getString(i))
}
buffer += map
}

buffer
}
finally {
conn.close()
}
}
}

看完代码，大家会发现JDBC中的JavaBean被Scala的case class所替代。另外，批量删除的代码我没写，基本上和JDBC中的批量删除没有差别。

总之，Scala对数据库的操作时调用Java的API来完成的，从这里也可以看到Scala与Java的无缝结合。我想这也许是为什么Scala编程书上没有相关数据库操作的内容的原因吧。

完整的代码（用的是MySQL数据）我已经托管到了GitHub上，并且源代码中还包含了JDBC的代码，二者可以对比。