lightdb.sql.connect.DataSourceConnectionManager.scala Maven / Gradle / Ivy

Go to download
package lightdb.sql.connect

import lightdb.doc.Document
import lightdb.sql._
import lightdb.transaction.Transaction

import java.sql.Connection
import javax.sql.DataSource

trait DataSourceConnectionManager extends ConnectionManager {
  protected def dataSource: DataSource

  private def openConnection(): Connection = {
    val c = dataSource.getConnection
    c
  }

  private def closeConnection(connection: Connection): Unit = {
    connection.commit()
    connection.close()
  }

  override def getConnection[Doc <: Document[Doc]](implicit transaction: Transaction[Doc]): Connection = {
    val state = getState
    synchronized {
      if (state.connection == null) {
        state.connection = openConnection()
      }
    }
    state.connection
  }

  override def currentConnection[Doc <: Document[Doc]](implicit transaction: Transaction[Doc]): Option[Connection] =
    Option(getState.connection)

  override def releaseConnection[Doc <: Document[Doc]](implicit transaction: Transaction[Doc]): Unit =
    currentConnection.foreach(closeConnection)
}