Play Framework/Scala example source code file (H2Database.scala)

This example Play Framework source code file (H2Database.scala) is included in my "Source Code Warehouse" project. The intent of this project is to help you more easily find Play Framework (and Scala) source code examples by using tags.

All credit for the original source code belongs to Play Framework; I'm just trying to make examples easier to find. (For my Scala work, see my Scala examples and tutorials.)

Play Framework tags/keywords

connection, h2database, int, long, play, play framework, r, string, string*, testtable, unit, utilities

The H2Database.scala Play Framework example source code

package anorm

import java.sql.{ DriverManager, Connection }
import scala.util.Random

trait H2Database {

  def withConnection[R](block: Connection => R) = {
    val dbName = "test" + Random.alphanumeric.take(6).mkString("")

    Class.forName("org.h2.Driver")
    val connection = DriverManager.getConnection("jdbc:h2:mem:" + dbName, "sa", "")
    try {
      block(connection)
    } finally {
      connection.close()
    }
  }

  case class TestTable(id: Long, foo: String, bar: Int)

  /** Create a simple 'test1' table for testing with. */
  def createTest1Table()(implicit conn: Connection): Unit = createTable("test1", "id bigint", "foo varchar", "bar int")

  /** Create a simple 'test2' table for testing with. */
  def createTest2Table()(implicit conn: Connection): Unit = createTable("test2", "id bigint", "foo varchar")

  private def createTable(name: String, columns: String*)(implicit conn: Connection): Unit = conn.createStatement().execute("create table " + name + " ( " + columns.mkString(", ") + ");")

}