org.r10r.sqlify.Database Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sqlify-core Show documentation

There is a newer version: 1.0.2

package org.r10r.sqlify;

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

public class Database {
  
  private final DataSource dataSource;

  public static interface Executable {
    T execute(Connection connection);
  }

  private Database(DataSource dataSource) {
    this.dataSource = dataSource;
  }

  public  T withConnection(Executable block) {
    return withConnection(true, block);
  }

  public  T withConnection(boolean autocommit, Executable block) {
    try (Connection connection = this.dataSource.getConnection()) {
      connection.setAutoCommit(autocommit);
      return block.execute(connection);
    } catch (Exception e) {
      throw new SqlifyException(e);
    }
  }

  public  T withTransaction(Executable block) {
    return withConnection(false, connection -> {
      try {
        T t = block.execute(connection);
        connection.commit();
        return t;

      } catch (Exception e1) {
        try {
          connection.rollback();
        } catch (Exception e2) {
          throw new SqlifyException(e2);
        }

        throw new SqlifyException(e1);
      }

    });
  }
  
  //////////////////////////////////////////////////////////////////////////////
  // Simplified constructor
  //////////////////////////////////////////////////////////////////////////////
  public static Database use(DataSource dataSource) {
    Database database = new Database(dataSource);
    return database;
  }
  
}