scalikejdbc.Tx.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalikejdbc-core_2.11 Show documentation

scalikejdbc-core

There is a newer version: 3.2.0-RC1

Show newest version

package scalikejdbc

import java.sql.{ SQLException, Connection }
import scala.util.control.Exception._
import scala.util.control.NonFatal

/**
 * DB Transaction abstraction.
 * @param conn connection
 */
class Tx(val conn: Connection) {

  /**
   * Begins this transaction.
   */
  def begin(): Unit = {
    conn.setAutoCommit(false)
    if (!GlobalSettings.jtaDataSourceCompatible) {
      conn.setReadOnly(false)
    }
  }

  /**
   * Commits this transaction.
   */
  def commit(): Unit = {
    try conn.commit() catch {
      case NonFatal(e) =>
        try conn.rollback() catch {
          case NonFatal(e2) => e.addSuppressed(e2)
        }
        throw e
    }
    conn.setAutoCommit(true)
  }

  /**
   * Returns is this transaction active.
   * Since connections from JTA managed DataSource should not be used as-is, we don't care autoCommit/readOnly props.
   * @return active
   */
  def isActive(): Boolean = GlobalSettings.jtaDataSourceCompatible || !conn.getAutoCommit

  /**
   * Rolls this transaction back.
   */
  def rollback(): Unit = {
    conn.rollback()
    ignoring(classOf[SQLException]) apply {
      conn.setAutoCommit(true)
    }
  }

  /**
   * Rolls this transaction back if this transaction is still active.
   */
  def rollbackIfActive(): Unit = {
    ignoring(classOf[SQLException]) apply {
      conn.rollback()
    }
    ignoring(classOf[SQLException]) apply {
      conn.setAutoCommit(true)
    }
  }

}