org.fluentjdbc.DbContextConnection Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fluent-jdbc Show documentation

A Java library used to execute JDBC statements and build SQL

There is a newer version: 0.5.3

package org.fluentjdbc;

import org.fluentjdbc.util.ExceptionUtil;

import java.sql.Connection;
import java.sql.SQLException;

public class DbContextConnection implements AutoCloseable {

    private ConnectionSupplier connectionSupplier;
    private Connection connection;
    private DbContext context;

    public DbContextConnection(ConnectionSupplier connectionSupplier, DbContext context) {
        this.connectionSupplier = connectionSupplier;
        this.context = context;
    }

    @Override
    public void close() {
        if (connection != null) {
            try {
                connection.close();
            } catch (SQLException e) {
                throw ExceptionUtil.softenCheckedException(e);
            }
        }
        context.removeFromThread();
    }

    Connection getConnection() {
        if (connection == null) {
            try {
                connection = connectionSupplier.getConnection();
            } catch (SQLException e) {
                throw ExceptionUtil.softenCheckedException(e);
            }
        }
        return connection;
    }

    public void commitTransaction() {
        if (connection != null) {
            try {
                connection.commit();
            } catch (SQLException e) {
                throw ExceptionUtil.softenCheckedException(e);
            }
        }
    }
}