io.github.mmm.orm.jdbc.session.JdbcSession Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mmm-orm-jdbc Show documentation
JDBC support for mmm-orm.
The newest version!
/* Copyright (c) The m-m-m Team, Licensed under the Apache License, Version 2.0
 * http://www.apache.org/licenses/LICENSE-2.0 */
package io.github.mmm.orm.jdbc.session;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.concurrent.Callable;

import io.github.mmm.orm.connection.DbConnectionData;
import io.github.mmm.orm.jdbc.connection.JdbcConnection;
import io.github.mmm.orm.jdbc.connection.JdbcConnectionPool;
import io.github.mmm.orm.jdbc.tx.JdbcTransactionExecutor;
import io.github.mmm.orm.source.DbSource;
import io.github.mmm.orm.spi.session.AbstractDbSession;
import io.github.mmm.orm.tx.DbTransaction;

/**
 * Database session data for a single transaction.
 */
public class JdbcSession extends AbstractDbSession implements DbTransaction {

  private static final ScopedValue SESSION_HOLDER = ScopedValue.newInstance();

  final Connection connection;

  final JdbcConnection jdbcConnection;

  /**
   * The constructor.
   *
   * @param jdbcConnection the {@link JdbcConnection}.
   * @param connectionData the {@link DbConnectionData}.
   */
  private JdbcSession(JdbcConnection jdbcConnection, DbConnectionData connectionData) {

    super(connectionData);
    this.jdbcConnection = jdbcConnection;
    this.connection = jdbcConnection.getConnection();
  }

  /**
   * @return the raw JDBC {@link Connection}.
   */
  public Connection getConnection() {

    return this.connection;
  }

  /**
   * @return the {@link JdbcConnection}.
   */
  public JdbcConnection getJdbcConnection() {

    return this.jdbcConnection;
  }

  @Override
  public boolean isOpen() {

    return this.jdbcConnection.isOpen();
  }

  @Override
  public DbSource getSource() {

    return this.jdbcConnection.getSource();
  }

  /**
   * @return the {@link JdbcSession} for the current transaction.
   */
  public static JdbcSession get() {

    JdbcSession session = SESSION_HOLDER.get();
    if (session == null) {
      throw new IllegalStateException("No session found - missing transactional context!");
    }
    return session;
  }

  /**
   * @param  type of the returned result.
   * @param connectionData the {@link DbConnectionData}.
   * @param task the {@link Callable} task to run in transaction.
   * @return the result of {@link Callable#call()}.
   * @see JdbcTransactionExecutor#doInTx(Callable)
   */
  public static  R doInTx(DbConnectionData connectionData, Callable task) {

    Connection connection = null;
    JdbcConnection jdbcConnection = null;
    JdbcConnectionPool connectionPool = (JdbcConnectionPool) connectionData.getPool();
    try {
      jdbcConnection = connectionPool.acquire();
      JdbcSession session = new JdbcSession(jdbcConnection, connectionData);
      R result = ScopedValue.where(SESSION_HOLDER, session).call(task);
      connection = session.getConnection();
      connection.commit();
      return result;
    } catch (Throwable t) {
      if (connection != null) {
        try {
          connection.rollback();
        } catch (SQLException e) {
          t.addSuppressed(e);
        }
      }
      if (t instanceof RuntimeException re) {
        throw re;
      } else if (t instanceof Error e) {
        throw e;
      }
      throw new IllegalStateException(t);
    } finally {
      connectionPool.release(jdbcConnection);
    }
  }

}